Moje pierwsze pytanie: Jestem zdezorientowany co do tego zachowania metody apply of groupby w Pandzie (0.12.0-4), wydaje się, że funkcja jest stosowana dwa razy do pierwszego wiersza ramki danych. Na przykład:

>>> from pandas import Series, DataFrame
>>> import pandas as pd
>>> df = pd.DataFrame({'class': ['A', 'B', 'C'], 'count':[1,0,2]})
>>> print(df)
   class  count  
0     A      1  
1     B      0    
2     C      2

Najpierw sprawdzam, czy funkcja groupby Działa ok i wydaje się być w porządku:

>>> for group in df.groupby('class', group_keys = True):
>>>     print(group)
('A',   class  count
0     A      1)
('B',   class  count
1     B      0)
('C',   class  count
2     C      2)

Potem próbuję zrobić coś podobnego używając apply na obiekcie groupby i otrzymuję wyjście pierwszego wiersza dwa razy:

>>> def checkit(group):
>>>     print(group)
>>> df.groupby('class', group_keys = True).apply(checkit)
  class  count
0     A      1
  class  count
0     A      1
  class  count
1     B      0
  class  count
2     C      2

Edit: @ Jeff podaje odpowiedź poniżej. Jestem gęsty i nie zrozumiałem tego od razu, więc oto prosty przykład, aby pokazać, że pomimo podwójnego wydruku pierwszej grupy w powyższym przykładzie, metoda apply działa tylko raz na pierwszej grupie i nie mutuje oryginalnej ramki danych: {]}

>>> def addone(group):
>>>     group['count'] += 1
>>>     return group

>>> df.groupby('class', group_keys = True).apply(addone)
>>> print(df)

      class  count
0     A      1
1     B      0
2     C      2

Ale przypisując zwracanie metody do nowego obiektu, widzimy, że działa ona zgodnie z oczekiwaniami:

>>> df2 = df.groupby('class', group_keys = True).apply(addone)
>>> print(df2)

      class  count
0     A      2
1     B      1
2     C      3