python - pandas.pivot_table : How to name functions for aggregation

我正在尝试使用几个聚合函数来透视 Pandas DataFrame，其中一些是 lambda。每列必须有一个不同的名称，以便通过多个 lambda 函数进行聚合。我尝试了一些我在网上找到的想法，但都没有奏效。这是最小的例子:

df = pd.DataFrame({'col1': [1, 1, 2, 3], 'col2': [4, 4, 5, 6], 'col3': [7, 10, 8, 9]})

pivoted_df = df.pivot_table(index = ['col1', 'col2'], values  = 'col3', aggfunc=[('lam1', lambda x: np.percentile(x, 50)), ('lam2', np.percentile(x, 75)]).reset_index()

错误是

AttributeError: 'SeriesGroupBy' object has no attribute 'lam1'

我尝试使用 dictionary ，它也会导致错误。有人可以帮忙吗？谢谢!

最佳答案

明确命名函数:

def lam1(x):
    return np.percentile(x, 50)

def lam2(x):
    return np.percentile(x, 75)

pivoted_df = df.pivot_table(index = ['col1', 'col2'], values  = 'col3',
                            aggfunc=[lam1, lam2]).reset_index()

然后将适本地命名您的聚合系列:

print(pivoted_df)

   col1  col2  lam1  lam2
0     1     4   8.5  9.25
1     2     5   8.0  8.00
2     3     6   9.0  9.00

pd.pivot_table 的 docs 解释了原因: