在Python中将两个数据框中的两列相乘并求和

Question

我有两个数据框，都有 6 行。我想将两个数据帧（每个 df 一个）中两个选定列中的值相乘，

result = sum(a * b for a, b in zip(list(df1['col1']), list(df2['col3'])))

我似乎没有得到我想要的。我在 Excel 中“手动”进行了计算（针对我的时间序列中的一个日期），这给了我预期的结果。所以我的问题是我是否做错了什么？

Answer 1

如果相同的行数和相同的索引简单相减，然后使用 sum：

result = (df1['col1'] * df2['col3']).sum()

如果可能不同的索引但相同的长度：

result  = (df1['col1'] * df2['col3'].to_numpy()).sum()

或使用 numpy.dot：

result = np.dot(df1['col1'],  df2['col3'])

---

如果可能的话，不同的长度和不同的索引：

result = (df1['col1'].reset_index(drop=True)
             .mul(df2['col3'].reset_index(drop=True), fill_value=1).sum()

Answer 2

你可以这样做：

import pandas as pd 
import numpy as np 

df1 = pd.DataFrame({'col1':[0, 1, 2, 3, 4, 5]})
df2 = pd.DataFrame({'col1':[0, 1, 2, 3, 4, 5]})

result = np.matmul(df1.col1, df2.col1)

这也会对乘法求和。

如果添加 []，您的公式也有效：

result = sum([a * b for a, b in zip(list(df1['col1']), list(df2['col1']))])

这会产生相同的结果。

Answer 3

您只需使用 pandas 抽象即可。

result = df['col1'] * df['col3']

如果您想获得这些结果值的总和，您可以这样做：

sum(results)