python - 如何对系列对象中每个类别的值求和？

Question

我有这个数据框:

我需要做的是按Category列对DataFrame进行分组，然后使用函数计算每个类别的总权重。这就是我到目前为止所做的...

def multiply(group, quantity, weight):
  x = group[quantity]
  y = group[weight]

  return x * y

print(df.groupby('Category').apply(multiply, 'Quantity',  'Weight (oz.)'))

我得到的结果是

现在，我如何按生成的 Series 对象的每个类别(即，背包、庇护所、 sleep 等)求和总重量？ p>

Answer 1

使用您的数据样本:

Item,Category,Quantity,Weight (oz.)
Sleeping Pad,Sleep,1,80.0
Sleeping Bag,Sleep,1,20.0
Spoon,Kitchen,1,0.87
Stove,Kitchen,1,20.0
Water Filter,Kitchen,1,1.8
Water Bottles,Kitchen,2,35.0

In [1]: df = pd.read_clipboard(sep=',', index_col=0)


In [2]: df.groupby('Category').apply(lambda x: (x['Quantity'] *x['Weight (oz.)']).sum())
Out[2]: 
Category
Kitchen     92.67
Sleep      100.00
dtype: float64