Pandas groupby 与 agg 一起使用不返回键列

Question

在我正在研究的一个项目中，我被迫使用 Pandas 1.1.5 版。 我正在尝试按操作进行分组，以便使用多个函数聚合变量：

import pandas as pd
import numpy as np

df = pd.DataFrame( { 
        "Name" : ["Alice", "Bob", "James", "Mallory", "Bob" , "Lucas", "Alice", "Bob", "James", "Mallory", "Bob" , "Lucas"] , 
        "Apples" : [22, 31, 35, 41, 27, 32, 64, 12, 59, 45, 65, 31] } )

apple_df = df.groupby('Name', as_index = False).agg(
    apple_avg = ('Apples', np.mean),
    apple_median = ('Apples', np.median),
    apple_count = ('Apples', np.count_nonzero)
)
apple_df

我期望名称列与其他聚合变量的结果如下：

但我得到以下信息：

此问题的任何已知错误和解决方法？

PS All 适用于 Pandas 1.3.0，但我不能在这个项目中使用它。

Answer 1

您可以尝试删除 as_index 参数并改为使用 a.reset_index() ：

apple_df = df.groupby('Name').agg(
    apple_avg = ('Apples', np.mean),
    apple_median = ('Apples', np.median),
    apple_count = ('Apples', np.count_nonzero)
).reset_index()