Python pandas pivot_table margins keyError

Question

考虑以下数据框：

test = pd.DataFrame({'A': [datetime.datetime.now(), datetime.datetime.now()], 'B': [1, 2]})

如果我使用如下所示的pivot_table那么一切都很好：

test.pivot_table(index = 'A', aggfunc = {'B': 'mean'}, margins = True)

但是，如果我执行以下操作，则无法设置margins = True （抛出错误KeyError: 'A' ）：

test.pivot_table(index = test['A'], aggfunc = {'B': 'mean'}, margins = True)

我真的很困惑。 假设我需要做类似下面的事情并且需要设置margin = True 。 那不可能吗？

test.pivot_table(index = test['A'].dt.year, aggfunc = {'B': 'mean'}, margins = True)

Answer 1

尝试：

test['Ax']=test['A'].dt.year

test.pivot_table(index = 'Ax' , aggfunc = 'mean', values='B', margins = True)

输出：

        B
Ax
2020  1.5
All   1.5

说明：如果您不传递values ，它将默认为df.columns （数据df.columns所有列，您正在旋转）。 https://github.com/pandas-dev/pandas/blob/v0.25.3/pandas/core/reshape/pivot.py#L87

所以从技术上讲，通过不传递任何values您将所有列传递到values ，但同时仅提供一个函数，所以这就是这个KeyError来源。