熊猫中的excel sumproduct

Question

我有一个DF：

Type   price   stock
a      2       2
b      4       1
b      3       3
a      1       2
a      3       1

我想获得的结果是：

Type   price*stock
a      2*2+1*2+3*1 = 9
b      4*1+3*3 = 13 

我可以在Excel中轻松地做到这一点，但是在熊猫中怎么样？我尝试了groupby功能，但仍然失败:(

Answer 1

第一个多列，然后汇总sum以提高性能：

df1 = df.price.mul(df.stock).groupby(df.Type).sum().reset_index(name='price*stock')
print (df1)
  Type  price*stock
0    a            9
1    b           13

另一个想法是一个具有多个值的克里特列列，然后将其汇总：

df1 = (df.assign(**{'price*stock': df.price.mul(df.stock)})
         .groupby('Type', as_index=False)['price*stock']
         .sum())
print (df1)
  Type  price*stock
0    a            9
1    b           13

Answer 2

groupby关于类型和将方程应用于每个组。

out = df.groupby("Type").apply(lambda x: sum(x["price"]*x["stock"])).reset_index(name="price*stock")

print(out)
>>  Type  price*stock
0   a         9
1   b         13

Answer 3

还：

df.groupby('Type').apply(lambda x:[email protected]).reset_index(name='price_stock')
 
  Type  price_stock
0    a            9
1    b           13

Answer 4

df.groupby('Type').apply(lambda x: x['price'].dot(x['stock'])).to_frame('sumproduct')



       sumproduct
Type            
a              9
b             13