Python pandas hub_table 多个时间索引

Question

我有以下数据框:

import pandas as pd
import datetime

df = pd.DataFrame({'T': [datetime.datetime.now(), datetime.datetime.now()], 'V': [1, 2]})

我想使用年份和月份作为索引进行旋转，我可以这样做:

df.pivot_table(index = [df['T'].dt.year, df['T'].dt.month], values = 'V')

但是，由于这给了我两个名为 T 的列，因此我无法设置 margins = True (引发错误:名称 T 出现多次，使用级别编号)。我该如何解决这个问题？理想情况下，我想在进行数据透视时手动命名列，这样我就有 Year 和 Month 而不是 T 和 T 。

有什么想法吗？

Answer 1

IIUC，您可以使用rename来重命名您的系列，

df.pivot_table(index = [df['T'].dt.year.rename('Year'),
                        df['T'].dt.month.rename('Month')], 
                values='V',
                aggfunc='sum', 
                margins=True)

输出:

              V
Year Month     
2020 1        3
All           3