как передать имя столбца в качестве параметра при групповом и применить

Question

В python-пандах, когда df.groupby('columnA').apply(myfunction), как передать columnA в функцию? Значения в столбце A будут использоваться для дальнейшей фильтрации данных.

Попробовал ниже и не получится

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

Заранее спасибо!

Answer 1

Вы можете получить доступ к столбцу, по которому вы группируете, через 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