如何根据多列数据在Pandas Python中的一个图中绘制多条线?

时间:2016-10-16 13:56:36

标签: python pandas matplotlib plot

我有一个包含3列的数据框,如下所示:

df['year'] = ['2005, 2005, 2005, 2015, 2015, 2015, 2030, 2030, 2030']
df['name'] = ['A', 'B', 'C', 'A', 'B', 'C', 'A', 'B', 'C']
df['weight'] = [80, 65, 88, 65, 60, 70, 60, 55, 65]

如何为A,B和C画一条线,它显示了这些年来它们的体重如何发展。所以我尝试了这个:

df.groupby("euro").plot(x="year", y="MKM")

然而,我得到了多个情节,这不是我想要的。我希望所有这些情节都在一个图中。

2 个答案:

答案 0 :(得分:2)

对于此特定示例,最佳解决方案是使用以下内容

df.set_index("year", inplace=True)
df.groupby("name")["weight"].plot(legend=True, xlabel="Year", ylabel="Weight")

enter image description here

答案 1 :(得分:1)

这会产生你想要的吗?

import matplotlib.pyplot as plt
fig,ax = plt.subplots()

for name in ['A','B','C']:
    ax.plot(df[df.name==name].year,df[df.name==name].weight,label=name)

ax.set_xlabel("year")
ax.set_ylabel("weight")
ax.legend(loc='best')

enter image description here