Панда: Продукт конкретных столбцов

Question

Нахождения произведения всех столбцов в dataframe легко:

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