將Pandas DataFrame中的列與DataFrame中的列列相結合

Question

請考慮以下DataFrame 。

n  v1 v2 v3 v4 v5
0   1  2  3  4  5
1   1  2  3  4  5
2   1  2  3  4  5

對於每一行，我想將v2 ， v3 ， v4的值添加到列表中，並將列表中的值與v5相乘，並將結果放入新的列v6 ，以便最終DataFrame如下所示的DataFrame ：

n  v1  v6
0   1  [10, 15, 20]
1   1  [10, 15, 20]
2   1  [10, 15, 20]

我怎樣才能在熊貓中實現這一目標？

Answer 1

你可以在一行中這樣做：

>>> df['v6'] = df[['v2', 'v3', 'v4']].mul(df['v5'], axis=0).values.tolist()
>>> df
   v1  v2  v3  v4  v5            v6
0   1   2   3   4   5  [10, 15, 20]
1   1   2   3   4   5  [10, 15, 20]
2   1   2   3   4   5  [10, 15, 20]

這會進行相關的列乘法，將v2 ， v3和v4值放到列表列表中（逐行）並創建新列v6 。