如何为a size（）列指定名称？

Question

我对groupby结果使用.size（）来计算每组中有多少项。

我希望在不手动编辑列名数组的情况下将结果保存为新列名，如何完成？

谢谢

这就是我尝试过的：

grpd = df.groupby(['A','B'])
grpd['size'] = grpd.size()
grpd

我得到的错误是：

TypeError: 'DataFrameGroupBy' object does not support item assignment (on the second line)

Answer 1

你需要^{}^{}-len的df和以前一样：

注意：

这里需要在groupby之后添加一列，否则会出现错误。因为^{}countNaNs也是，所以使用什么列并不重要。所有列工作相同。

import pandas as pd

df = pd.DataFrame({'A': ['x', 'x', 'x','y','y']
                , 'B': ['a', 'c', 'c','b','b']})
print (df)
   A  B
0  x  a
1  x  c
2  x  c
3  y  b
4  y  b

df['size'] = df.groupby(['A', 'B'])['A'].transform('size')
print (df)
   A  B  size
0  x  a     1
1  x  c     2
2  x  c     2
3  y  b     2
4  y  b     2

如果需要在聚合df-len的df中设置列名，则显然与之前不同：

import pandas as pd

df = pd.DataFrame({'A': ['x', 'x', 'x','y','y']
                , 'B': ['a', 'c', 'c','b','b']})
print (df)
   A  B
0  x  a
1  x  c
2  x  c
3  y  b
4  y  b

df = df.groupby(['A', 'B']).size().reset_index(name='Size')
print (df)
   A  B  Size
0  x  a     1
1  x  c     2
2  y  b     2