當我使用pandas DataFrame groupby然后應用用戶定義的函數時，調用了多少個函數？

Question

我有一個熊貓DataFrame：

df = pd.DataFrame({'A':['a', 'a', 'a', 'a', 'a'], 'B':[1, 2, 3, 4, 5]})

和用戶定義的函數：

def f(df):
    return df.max() - df.min()

當我運行以下代碼時：

df.groupby('A').apply(f)

函數f將被調用多少次？ 我測試了這段代碼，發現f將被調用4次，有人可以解釋嗎？

Answer 1

應該將其稱為ngroup times，如果A中有兩個group，它將被稱為兩次。 另外，您可以使用np.ptp

df.groupby('A').B.apply(np.ptp)
Out[52]: 
A
a    4
Name: B, dtype: int64