pandas Group通过绘制每个组

Question

我有一些数据，我想从中提取不同产品（和）的收入时间序列（Dollars不同Day地点不同日期的总和） 。Wherexy

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()