如何将多列转换为哑变量？对industryCode列执行独热编码并保留其他列

原始数据 #

需求：对industryCode列执行独热编码（One-Hot Encoding），同时保留date和windCode列。

实现代码（Python Pandas） #

import pandas as pd

# 构造原始数据
df = pd.DataFrame({
    'date': ['2022.08.01', '2022.08.02', '2022.08.03'],
    'windCode': ['000001.SZ', '000002.SZ', '000003.SZ'],
    'industryCode': ['1101', '1102', '1103']
})

# 执行独热编码，保留其他列
# 使用prefix=''和prefix_sep=''让编码后的列名直接为industryCode的取值
df_encoded = pd.get_dummies(df, columns=['industryCode'], prefix='', prefix_sep='')

# 输出结果
print(df_encoded)

转换后结果 #

内容的提问来源于stack exchange，提问作者Huang WeiFeng