如何在Pandas DataFrame中添加特征间的乘积新特征?
Pandas添加两两乘积特征的实现方法
方法一:手动逐个创建(适合特征数量少的场景)
如果你的特征数量不多(比如示例中的4个),直接手动为每个乘积组合创建新列即可,操作直观且容易理解:
import pandas as pd # 假设你的DataFrame名为df,包含X_1到X_4列 df['X_1*X_2'] = df['X_1'] * df['X_2'] df['X_1*X_3'] = df['X_1'] * df['X_3'] df['X_1*X_4'] = df['X_1'] * df['X_4'] df['X_2*X_3'] = df['X_2'] * df['X_3'] df['X_2*X_4'] = df['X_2'] * df['X_4'] df['X_3*X_4'] = df['X_3'] * df['X_4']
执行后原特征列会完整保留,新的乘积特征会追加到DataFrame末尾。
方法二:自动遍历组合(适合特征数量多的场景)
如果后续特征数量增加,手动写代码会很繁琐,用itertools.combinations可以自动生成所有两两特征的组合,批量创建乘积特征:
import pandas as pd from itertools import combinations # 定义需要生成乘积的特征列列表 feature_cols = ['X_1', 'X_2', 'X_3', 'X_4'] # 遍历所有两两特征组合,生成新列 for col1, col2 in combinations(feature_cols, 2): new_col = f"{col1}*{col2}" df[new_col] = df[col1] * df[col2]
这种方式会自动处理所有不重复的两两组合,无需手动编写每个乘积逻辑,同样保留原特征列。
内容的提问来源于stack exchange,提问作者rose
相关产品推荐
相关产品推荐

