Groupby 多列计算大小并计算 Pandas 中另一列的平均值

ah bon 发表于 Dev

是这样吗

我有一个数据框如下：

   index    col1    col2   col3  col4                           col5
0      0  Week_1   James   John     1                  when and why?
1      1  Week_1   James   John     3             when and why? How?
2      2  Week_2   James   John     2  How far is it? Are you going?
3      3  Week_2    Mark    Jim     3              Do you know when?
4      4  Week_2  Andrew  Simon     3                     What time?
5      5  Week_2  Andrew  Simon     6                     What time?

我怎么能GROUPBYcol2和col3然后计算平均值和计数的数量col2和col3？

df.groupby(['col2','col3'], as_index=False).agg({'col4':'mean'}).reset_index()

输出：

   index    col2   col3  col4
0      0  Andrew  Simon   4.5
1      1   James   John   2.0
2      2    Mark    Jim   3.0

df.groupby(['col2','col3']).size().reset_index()

输出：

     col2   col3  0
0  Andrew  Simon  2
1   James   John  3
2    Mark    Jim  1

我怎么能得到这样的结果？谢谢。

   index    col2   col3  mean  count
0      0   James   John   2.0      3
1      3    Mark    Jim   3.0      1
2      4  Andrew  Simon   4.5      2

艾伦

您可以使用 groupby 和 agg（可能需要使用 pandas 0.25+）。

(
    df.groupby(['col2','col3'])
    .agg(index=('index', 'first'),
         mean=('col4', 'mean'),
         count=('col4', 'size'))
    .reset_index()
    .sort_values(by='index')
)

    col2    col3    index   mean    count
1   James   John    0       2.0     3
2   Mark    Jim     3       3.0     1
0   Andrew  Simon   4       4.5     2

本文收集自互联网，转载请注明来源。

如有侵权，请联系 [email protected] 删除。