python - 获取 Pandas 中 groupby 操作的大小

Question

我一直在对数据框执行 groupby 操作，我根据“名称”列将列聚合在一起:

Name | As | Bs | Cs   |  Note
Mark   3     4     7     Good
Luke   2     1     12    Well
Mark   5     6     8     Ok
John   1     18    3     Great

因此在这种情况下，使用以下代码将带有“Mark”的行聚合到 A、B 和 C 列中:

temp_df = temp_df.groupby(['Name'], as_index=False).agg({'As': np.sum, 'Bs': np.sum,'Cs': np.sum})

我需要添加的一件事是计算“名称”中具有相同值的行数。这会给我这样的输出:

Name | As | Bs | Cs   |  Note   | Count
Mark   8     10    15    Good      2
Luke   2     1     12    Well      1
John   1     18    3     Great     1

如何修改上面的代码行来完成我需要的事情？

Answer 1

创建组并进行聚合:

the_group = temp_df.groupby(['Name'], as_index=False)
temp_df = the_group.agg({'As': np.sum, 'Bs': np.sum,'Cs': np.sum})

然后根据the_group计算大小

temp_df['count'] = the_group.count()['Note']

给出:

   Name  Cs  As  Bs  count
0  John   3   1  18      1
1  Luke  12   2   1      1
2  Mark  15   8  10      2

编辑:

如评论中所建议，如果数据包含 NaN，则使用 size() 会更安全:

temp_df['count'] = the_group.size().reset_index()[0]