python - pandas Group通过绘制每个组

Question

我有一些数据，我想从中提取收入的时间序列(Dollar 的总和 Day 不同地点 Where) 用于不同的产品(x 和 y)。

import pandas as pd

#Create data
data = {'Day': [1,1,2,2,3,3],
        'Where': ['A','B','A','B','B','B'],
        'What': ['x','y','x','x','x','y'],
        'Dollars': [100,200,100,100,100,200]}

index = range(len(data['Day']))

columns = ['Day','Where','What','Dollars']


df = pd.DataFrame(data,  index=index, columns=columns)
df

为此，我按Day 和What 对数据进行分组，然后对Dollars 求和:

#Group by Day and What and sum Dollars (for each Where)
print(df.groupby(['Day', 'What'])['Dollars'].sum())

现在，我想为 x 和 y 制作一个时间序列，如下所示:

我尝试了以下方法，但显然不起作用:

items = df.What.unique()

ax = plt.figure()
for item in items:
    df_tmp = df[['Day']][df.What == item]
    plt.plot(df_tmp['Day'],df_tmp,'.-',label=item)

有人可以帮我指明正确的方向吗？有没有更快的方法来获得正确的结果？

Answer 1

IIUC、unstack 和绘图:

(df.groupby(['Day', 'What'])['Dollars']
   .sum()
   .unstack('What', fill_value=0)
   .plot())
plt.show()