python-3.x - pivot_table 在 Pandas 中返回 KeyError

Question

当我在 pandas 中使用 pivot_table 时遇到一个关键错误，我无法解释：

我的数据：

df1
make    body-style  engine-size
0   alfa-romero convertible 130
1   alfa-romero convertible 130
2   alfa-romero hatchback   152
3   audi    sedan   109
4   audi    sedan   136
5   audi    sedan   136
6   audi    sedan   136
7   audi    wagon   136
8   audi    sedan   131
10  bmw sedan   108

编码：

pd.pivot_table(df1, columns = ['make', 'body-style'], \
               margins = True, aggfunc = {'engine-size' : 'mean', 'make' : 'count'})
KeyError: 'make'

Answer 1

您可以使用agg

df.groupby(['make', 'body-style']).agg({'engine-size' : 'mean', 'make' : 'count'})
Out[128]: 
                         make  engine-size
make        body-style                    
alfa-romero convertible     2        130.0
            hatchback       1        152.0
audi        sedan           5        129.6
            wagon           1        136.0
bmw         sedan           1        108.0