Как преобразовывать категориальные данные с закодированными значениями?

Question

[mva178]

Подскажите,пожалуйста.
У меня есть датасэт, где одна из переменных (доход семьи) имеет значения от 0 до 5, т.е. 0 - доход до 5000 руб. и т.д. Как можно преобразовать данные для подачи инпутом в нейронку или Svm?
Сделать 6 переменных со значениями 0 и 1 подойдет? есть альтернативы?

Comments

[Станислав Макаров]

А что нужно получить на выходе нейросети? Сколько планируется выходов и какие значения там ожидаются?

[mva178]

Необязательно нейросеть,пока остановился на деревьях,но переделывать руками 1000 значений в 1000 переменных с бинарными значениями не осилил

Answer 1

[doktr]

Подойдет функция из pandas:
pd.get_dummies(df['n_school'])