[Python] pandasにてgroupby結果にapplyした際に、なぜ1周目が2回実行されるのか。

Question

実行環境：
OS：Windows10 pro
Python 3.7.3
pandas 0.24.2

掲題の通りですが、まずサンプルコードを記載します。

dummy = {
    'id': ['101', '102', '103', '104', '105'],
    'grp_key': ['aa', 'aa', 'bb', 'cc', 'aa']
}
df = pd.DataFrame(dummy)
print(df)
↓
    id grp_key
0  101      aa
1  102      aa
2  103      bb
3  104      cc
4  105      aa

def func(_df):
    print(_df)

df.groupby('grp_key').apply(func)

上記コードを実行したとき、
funcメソッドの中のprintでは以下のように出力されます。

【1回目】
id grp_key
0 101 aa
1 102 aa
4 105 aa

【2回目】
id grp_key
0 101 aa
1 102 aa
4 105 aa

【3回目】
id grp_key
2 103 bb

【4回目】
id grp_key
3 104 cc

df.groupby('grp_key').groupsのDataFrame配列分？の要素がapplyのメソッドに渡されてくることを期待していましたが、
誤りがありますか？

ご教示お願い致します。

Answer 1

最初のグループに対して、apply で適用する関数が2回呼ばれるのは pandas の仕様です。
バグではありません。

The apply method over a group seems to visit the first group twice · Issue #2656 · pandas-dev/pandas · GitHub

Comments

[sakichan]

tiitoi 様

ご回答いただきありがとうございます。
張っていただいたリンクを確認し、Pandasの仕様とのことで理解しました。
apply関数内には、この動作を踏まえたコーディングをしたいと思います。