使用pandas绘制带有年数的数据帧

Question

我的数据框看起来像这样：

country,2010,2011,2012
Afghanistan,1,2,3
Belize,5,3,2
England,3,3,4

我想绘制此数据框，以便每个国家/地区都由折线图中的一条线表示，其中x轴代表年份。

我该怎么做？

Answer 1

我认为您可能需要重塑数据框：

%matplotlib inline
from io import StringIO
import pandas as pd
import matplotlib as mpl
mpl.rc("figure", figsize=(8,6))

data = """
country,2010,2011,2012
Afghanistan,1,2,3
Belize,5,3,2
England,3,3,4
"""

df = pd.read_csv(StringIO(data))
reshaped_df = pd.melt(df, id_vars=["country"], var_name="year")
df = reshaped_df.pivot(index='year', columns='country', values='value')

df.plot()