Pandas 数据帧中列之间的点积

Question

我需要计算熊猫数据帧列之间的点积并将结果存储在不同的列中。如何做到这一点。

column1 column2 column3    column4 column5 column6
1       2       3           1      2       3
4       5       6           4      5       6

假设我需要计算点积 列 1、列 2、列 3 和列 4 、列 5、列 6 即

column1 * column4 + column2 * column5 + column3 * column6

并将结果存储在其他列中，假设 df['dot']。

那么如何做到这一点呢？

Answer 1

df['dot'] = (df[['col1','col2','col3']].values * df[['col4,'col5','col6']].values).sum(axis=1)