python - Pandas GroupBy 列名

Question

我有一个数据框，我们可以通过代理

df = pd.DataFrame({'a':[1,0,0], 'b':[0,1,0], 'c':[1,0,0], 'd':[2,3,4]})

还有一个类别系列

category = pd.Series(['A', 'B', 'B', 'A'], ['a', 'b', 'c', 'd'])

我想获得 df 的列的总和，这些列分为“A”、“B”类别。可能是这样的:

result = df.groupby(??, axis=1).sum()

返回

result = pd.DataFrame({'A':[3,3,4], 'B':[1,1,0]})

Answer 1

在列上使用 groupby + sum(axis=1 在这里很重要):

df.groupby(df.columns.map(category.get), axis=1).sum()

   A  B
0  3  1
1  3  1
2  4  0