python - Python Pandas 按一列分组

Question

假设我在 Python 中有一个这样的数据框；

varX varY varZ
01   A     34
01   B     26
02   A     18
02   B     50

如何通过 varX 对该数据框进行分组并获取第三列（varZ）的总和，以便我们可以创建一个如下所示的新数据框；

varX varZ
01   60
02   68

我已经尝试了以下但没有奏效。

df_ = df.groupby(["varX"]).sum()

任何帮助深表感谢。谢谢！

Answer 1

尝试sum为varZ列做：

print(df.groupby("varX")['varZ'].sum())

正是你想要做的as_index=False：

print(df.groupby("varX",as_index=False)['varZ'].sum())