pandas：如何通过在 DataFrame 中进行分组来获取行的总和？

Question

我是数据科学新手，目前正在 Jupyter 笔记本上使用 Pandas 库。 对不起我的英语不好。

A,1,5,9
B,2,6,3
A,3,7,2
B,4,8,1

如何在创建 DataFrame 后将上述 CSV 值分组并添加内容？ 我想要这样的 output 。

    A   B
0   4   6
1   12  14
2   11  4

提前致谢：）

Answer 1

您可以使用df.pivot_table

df
   0  1  2  3
0  A  1  5  9
1  B  2  6  3
2  A  3  7  2
3  B  4  8  1

df.pivot_table(columns=0,aggfunc='sum').rename_axis(columns=None)
    A   B
1   4   6
2  12  14
3  11   4

Answer 2

这是带有转置的 groupby + sum：

print(df.groupby(0).sum().T.rename_axis(columns=None))

---

    A   B
1   4   6
2  12  14
3  11   4

注意：将df.groupby(0)中的 0 替换为第一列的实际列名