python - 列上的 Pandas Multiindex Groupby

Question

无论如何在 Multiindex 中的列上使用 groupby。我知道你可以在行上有很好的documentation在这方面。但是我似乎无法在列上分组。我唯一的解决方案是转置数据框。

#generate data (copied from pandas example)
arrays=[['bar', 'bar', 'baz', 'baz', 'foo', 'foo', 'qux', 'qux'],['one', 'two', 'one', 'two', 'one', 'two', 'one', 'two']]
tuples = list(zip(*arrays))
index = pd.MultiIndex.from_tuples(tuples, names=['first', 'second'])
df = pd.DataFrame(np.random.randn(3, 8), index=['A', 'B', 'C'], columns=index)

现在我将尝试对失败的列进行分组

df.groupby(level=1)
df.groupby(level='first')

但是行转置是可行的

df.T.groupby(level=1)
df.T.groupby(level='first')

那么有没有不转置的方法呢？

Answer 1

需要在groupby方法中指定坐标轴:

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

或者如果你的意思是 groupby 级别 0:

df.groupby(level = 0, axis = 1).sum()