在多索引数据框 pandas 中进行聚合

Question

我正在寻求与此多索引数据帧相关的帮助

import numpy as np
import pandas as pd

array = [np.array(['jan','jan','feb','feb','mar','mar']), np.array(['food','rent','food','rent','food','rent'])]

df = pd.DataFrame(np.random.randint(0,high=100,size=(6,1)),index=array,columns=['expense'])

我正在尝试总结整个数据框中“食物”和“租金”的数量 - 最简单的方法是什么？

Answer 1

您可以groupby 索引级别（food 和rent 处于级别 1 索引）：

df.groupby(level=1).sum()

#      expense
#food      166
#rent      161