Преобразование атрибута, имеющего значения в строковой форме, в вектор с использованием python

Question

Я должен подготовить данные для обучения модели линейной регрессии.В данных есть атрибут «Пол», имеющий три значения «М», «F» и «I».Как я могу преобразовать эти значения в соответствующую (1,0,0), (0,1,0), (0,0,1) векторную форму?

Answer 1

Просто используйте get_dummies()

import pandas as pd
df=pd.DataFrame({'Gender':['M','F','I']}) 
pd.get_dummies(df)

Gender_F    Gender_I    Gender_M
0           0           1
1           0           0
0           1           0