对一组 groupby 对象中的值求和？

Question

我正在尝试为我分组的每个条目总结 groupby 对象中列的值。

假设我有一个这样的 df：

Letters Numbers Items Bool
A       1       lamp  1
B       2       glass 1
B       2       table 1
C       5       pic   0

我对字母进行分组，然后想知道字母组中布尔值的总和。我该怎么做？我一直在努力

df_new = df.groupby('letters').bool.sum()

...

df_new = df.groupby('letters').sum('bool')

以及其他变体...

最后，我想得到一个向量，其中包含每个字母组之和的值。例如，它将是[1,2,0]。

Answer 1

你真的很亲密！给定

>>> df
  Letters  Numbers  Items  Bool
0       A        1   lamp     1
1       B        2  glass     1
2       B        2  table     1
3       C        5    pic     0

您可以对所有内容求和并取您想要的列：

>>> # slower
>>> df.groupby("Letters").sum()["Bool"] # sum everything, select Bool
Letters
A          1
B          2
C          0
Name: Bool, dtype: int64

或者更好的是，只取你想要的列并求和：

>>> df.groupby("Letters")["Bool"].sum() # select Bool, sum it
Letters
A          1
B          2
C          0
Name: Bool, dtype: int64

我更喜欢使用Series，因为您可以使用它做更多事情，但如果您愿意，可以使用list 或.tolist() 将其转换为list。