python - 如何在 pandas 数据框上应用 groupby 两次？

Question

我有 pandas 数据框，其中包含“年”、“月”和“交易 ID”列。我想获取每年每个月的交易数量。例如，我的数据如下:

year: {2015,2015,2015,2016,2016,2017}
month: {1,  1,   2,   2,   2,    1}
tid: {123,  343, 453, 675, 786, 332}

我希望获得输出，以便每年我都能获得每月的交易数量。对于 2015 年的 ex，我将得到输出:

month: [1,2]
count: [2,1]

我使用了groupby('year')。但之后我如何才能获得每月的交易数量。

Answer 1

您需要groupby按两列 - 年 和 月，然后汇总 size :

year = [2015,2015,2015,2016,2016,2017]
month =  [1,  1,   2,   2,   2,    1]
tid = [123,  343, 453, 675, 786, 332]

df = pd.DataFrame({'year':year, 'month':month,'tid':tid})
print (df)
   month  tid  year
0      1  123  2015
1      1  343  2015
2      2  453  2015
3      2  675  2016
4      2  786  2016
5      1  332  2017

df1 = df.groupby(['year','month'])['tid'].size().reset_index(name='count')
print (df1)
   year  month  count
0  2015      1      2
1  2015      2      1
2  2016      2      2
3  2017      1      1