Python pandas:向我的数据框中添加一列计算变量的列

问题描述

我有一个像这样的数据框gt":

I have a dataframe 'gt' like this:

org     group
org1      1
org2      1
org3      2
org4      3
org5      3
org6      3

我想将列计数"添加到 gt 数据框以计算组的成员数量，预期结果如下:

and I would like to add column 'count' to gt dataframe to counts number member of the groups, expected results like this:

org     group   count
org1      1       2
org2      1       2
org3      2       1
org4      3       3
org5      3       3
org6      3       3

我知道如何对组中的每一项进行计数，但不知道如何使所有组项的计数重复，这是我使用的代码:

I know how to do it per one item of the group, but do not know how to make the count repeated for all of the group items, here is the code I have used:

gtcounts = gt.groupby('group').count()

有人可以帮忙吗?

推荐答案

调用 transform 这将返回一个与原始 df 对齐的系列:

Call transform this will return a Series aligned with the original df:

In [223]:

df['count'] = df.groupby('group')['group'].transform('count')
df
Out[223]:
    org  group  count
0  org1      1      2
1  org2      1      2
2  org3      2      1
3  org4      3      3
4  org5      3      3
5  org6      3      3

这篇关于Python pandas:向我的数据框中添加一列计算变量的列的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持！