通过对数据框中的列进行分组来绘制折线图

Question

我有一个 csv 文件，其中包含我按月份对信息进行分组的数据，然后使用 cumsum 将月份的运行总计计算到一个数据框中。

使用此代码：

df = df.sort_index(sort_remaining=True).sort_values('months')
df['value'] = df.groupby('months')['value'].cumsum()

EXCEL 中的输出示例，但我的 DF 与 1000 行相同：

我现在想绘制一个图表，将月份分组并绘制每个值，所以基本上我将有 12 条绘制线显示值如何随着时间的推移而升高或降低。

输出图将如下图所示，显示每个月的累积和：

#

感谢@jezrael，它现在可以工作了。下面是剧情

Answer 1

我认为需要 pivot 和 rename 来代替数字名称，对于新索引值使用 cumcount：

d = {1: 'Jan', 2: 'Feb', 3: 'Mar', 4: 'Apr', 5: 'May',
     6 : 'Jun',7: 'Jul', 8: 'Aug', 9: 'Sep', 10: 'Oct', 11: 'Nov', 12: 'Dec'}

g = df.groupby('months').cumcount()
pd.pivot(index=g, columns=df['months'], values=df['value']).rename(columns=d).plot()

详情：

print(pd.pivot(index=g, columns=df['months'], values=df['value']).rename(columns=d))
months    Jan   Feb   Mar   Apr
0        50.0   2.0  10.0   5.0
1        80.0   3.0  16.0  20.0
2       120.0   8.0  31.0  40.0
3       140.0  11.0  34.0  50.0
4         NaN  15.0  43.0  75.0

编辑：

仅定义几个月的情节使用subset：

months = ['Mar','Apr']
g = df.groupby('months').cumcount()
pd.pivot(index=g, columns=df['months'], values=df['value']).rename(columns=d)[months].plot()

或者通过boolean indexing和isin过滤输入DataFrame中的月份：

df = df[df['months'].isin([3,4])]
g = df.groupby('months').cumcount()
pd.pivot(index=g, columns=df['months'], values=df['value']).rename(columns=d)[months].plot()