ivot_table返回熊猫中的关键错误

Question

我在熊猫中使用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