组和应用时如何传递列名作为参数

Question

在python熊猫中，df.groupby('columnA').apply(myfunction)何时执行操作，如何将columnA传递给函数？ columnA中的值将用于进行进一步的数据过滤。

在下面尝试过，将无法正常工作

df.groupby('columnA').apply(myfunction,('columnA'))

谢谢！

Answer 1

您可以通过{p>通过GroupBy访问要分组的列

df.groupby('columnA')['columnA'].apply(myfunction)

---

最小可验证示例

def myfunc(x):
    return x.sum()

df = pd.DataFrame({'A': list('abbac'), 'B': range(5)})
df
   A  B
0  a  0
1  b  1
2  b  2
3  a  3
4  c  4

df.groupby('A')['A'].apply(myfunc)    
A
a    aa
b    bb
c     c
Name: A, dtype: object