汇总数据框架的平均值并绘制线条图

时间:2018-06-04 08:23:16

标签: python matplotlib

我试图绘制一个组合3个数据帧的折线图,并计算不同参数的列平均值。我能够生成线图,但我不知道如何调整x和y边距。标签也包括“正确”字样。当我只想将键值作为图例时使用键值。我承认代码可能看起来很乱,但我不确定如何让它工作。任何提示非常感谢。

df1['id'] = df1['id'].map({0: 0.1, 1: 0.2, 2:0.5})
df2['id'] = df2['id'].map({0: 0.1, 1: 0.2, 2:0.5})
df3['id'] = df3['id'].map({0: 0.1, 1: 0.2, 2:0.5})

df1 = df1.groupby(['id']).agg({'correct': 'mean',}).reset_index()
df2 = df2.groupby(['id']).agg({'correct': 'mean',}).reset_index()
df3 = df3.groupby(['id']).agg({'correct': 'mean',}).reset_index()
df1['Key'] = 'A'
df2['Key'] = 'B'
df3['Key'] = 'C'

DF = pd.concat([df1,df2,df3],keys=['A','B','C'])

DFGroup = DF.groupby(['id','Key'])

DFGPlot = DFGroup.sum().unstack('Key').plot(kind='line')

plt.show()

0 个答案:

没有答案