python - Pandas :特定列的产品

Question

查找数据框中所有列的乘积很容易:

df['Product'] = df.product(axis=1)

如何指定要包含在乘积操作中的列名称(不是列编号)？

来自 DataFrame.product() 的帮助页面，我不确定这是否可能。

Answer 1

您可以使用 df[[colname1, colname2, colname3...]] 语法来选择您想要的列，然后调用 .product:

>>> df = pd.DataFrame({"A": [2,2], "B": [3,3], "C": [5,5]})
>>> df
   A  B  C
0  2  3  5
1  2  3  5

[2 rows x 3 columns]
>>> df[["A", "C"]].product(axis=1)
0    10
1    10
dtype: int64