python - 在 pandas 中使用 sumproduct 或 dot 函数

Question

我有如下2张表，df和name_df，其中人数不限2，可以增加。在名称列(即 John 和 Mary)下，1 表示他/她消费了该项目，而 0 表示没有消费:

df
item   price John  Mary
apple  2     1     1
mango  5     0     1

name_df
name
John
Mary

我想计算每个人的sumproduct并将列连接到name_df列。我希望结果看起来像这样:

name_df
name   sumproduct
John   2
Mary   7

我的主要问题是找出如何使用 python/pandas 计算 sumproduct。我知道 .dot() 函数会得到我想要的结果，但我如何才能将这些结果附加到每个人？

Answer 1

您可以使用 name_df 来确定 id_vars 和 groupby。

---

u = df.melt(df.columns ^ name_df['name'])

u.price.mul(u.value).groupby(u.variable).sum()

variable
John    2
Mary    7
dtype: int64