#!/usr/bin/env python
# coding: utf-8

# In[1]:


get_ipython().run_line_magic('matplotlib', 'inline')
import matplotlib.pyplot as plt
plt.style.use("ggplot")


# In[2]:


from pandas_datapackage_reader import read_datapackage
from shortcountrynames import to_name


# In[3]:


emissions = read_datapackage("https://github.com/openclimatedata/global-carbon-budget", "territorial-emissions")


# In[4]:


emissions.head()


# In[5]:


emissions = emissions.drop("Source", axis=1)


# In[6]:


unit = "MtC"


# In[7]:


ax = emissions.loc["CHN"].plot()
emissions.loc["USA"].plot(ax=ax)
plt.legend([to_name("CHN"), to_name("USA")])
plt.ylabel(unit);


# In[8]:


consumption = read_datapackage("https://github.com/openclimatedata/global-carbon-budget", "consumption-emissions")


# In[9]:


consumption.head()


# In[10]:


unit_consumption = "MtC"


# In[11]:


parties = ["USA", "CHN"]
for code in parties:
    ax = consumption.loc[code].plot()
    emissions.loc[code].plot(ax=ax)
    assert unit == unit_consumption
    plt.title(to_name(code))
    plt.ylabel(unit)


# In[ ]: