我需要为线性回归模型的训练准备数据。数据中有一个名为’Gender’的属性，它有三个值’M’、’F’和’I’。我如何将这些值转换为相应的(1,0,0)、(0,1,0)和(0,0,1)的向量形式？

回答：

只需使用 get_dummies()

import pandas as pddf=pd.DataFrame({'Gender':['M','F','I']}) pd.get_dummies(df)Gender_F    Gender_I    Gender_M0           0           11           0           00           1           0