pandas 数据框中各列之间的点积

Question

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

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

假设我需要计算点积列 1、列 2、列 3 和列 4 , column5, column6 即

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)

dot product between columns in the pandas dataframe