Как я могу закодировать категориальный столбец с кодами, которые я хочу?

Question

У меня есть такой фрейм:фиктивный пример из sklearn:

from sklearn.preprocessing import OneHotEncoder
enc = OneHotEncoder(handle_unknown='ignore')
X = [[1,'Male'], [ 3,'Female']]
enc.fit(X)

Я получаю следующую ошибку:

 ValueError: could not convert string to float: 'Male'

Заранее спасибо.

Answer 1

вы можете использовать map

gender = {'male':1,'female':3}
df.gender.map(gender)