使用pandas为DataFrame车型字段生成0/1二值标识列的最佳方法

最优实现方案 #

直接使用pandas内置的Series.str.get_dummies()方法即可，该方法天生支持对分隔符分隔的多值字段生成哑变量（0/1矩阵），无需手动枚举所有车型，代码简洁且执行效率更高。

示例代码：

import pandas as pd

# 假设你的原始DataFrame名称为df，只需执行以下一行代码即可完成需求
df = pd.concat([df, df['cars'].str.get_dummies(sep=',')], axis=1)

方案说明：

• df['cars'].str.get_dummies(sep=',')会自动识别cars列中所有用逗号分隔的车型，为每个车型生成单独的列，对应行包含该车型则值为1，否则为0
• 用pd.concat将生成的哑变量矩阵和原始DataFrame按列拼接，即可得到目标结构的结果
• 相比多条件np.where方案，无需手动维护所有车型的判断条件，当车型数量较多时优势尤其明显，后续新增车型也无需修改代码逻辑

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