如何基于列映射减去两个数据帧？

Question

我试图从dfb中减去dfa，其中列映射基于第三个映射数据帧。在本例中，应从x1中减去B，从x2中减去A。

这可以通过循环和其他一些肮脏的方法来实现，但我想知道是否有更简洁的方法来实现这一点。

数据帧a

数据帧b

数据帧c

所需结果

Answer 1

在数据帧上设置日期索引，从第三个数据帧创建一个系列；使用序列映射映射df1的列，最后执行减法：

df1 = df1.set_index('date')

df2 = df2.set_index('date')

df3 = df3.set_index('To').From

df1.columns = df1.columns.map(df3)

df2 - df1

             x1   x2
date
12/31/2019  0.6  0.7
12/31/2020 -0.2  0.4

Answer 2

您可以使用rename临时重命名列并进行减法。假设日期是您的索引：

a - b.rename(columns=c.set_index('From')['To'])

输出：

               A    B
date
12/31/2019  -0.7 -0.6
12/31/2020  -0.4  0.2

Answer 3

在减法之前使用merge：

tmp = pd.merge(dfa, dfb, on='date')
dfb[dfc['From']] = tmp[dfc['From']].values - tmp[dfc['To']].values
print(dfb)

# Output:
         date   x1   x2   x3
0  12/31/2019  0.6  0.7  0.9
1  12/31/2020 -0.2  0.4  1.2