Как получить pandas Series фиктивное представление одного значения - PullRequest
0 голосов
/ 07 ноября 2018

Предположим, у меня есть 5 разных категорий

categories = {'a', 'b', 'c', 'd', 'e'}

Можно ли использовать Pandas 'get_dummies для получения фиктивного представления одного элемента? А именно, предположим, у меня есть

element = 'a'

Становится

Series({
 'a' : 1, 
 'b' : 0,
 'c' : 0,
 'd' : 0,
 'e' : 0,
 })

1 Ответ

0 голосов
/ 07 ноября 2018

Пожалуйста, проверьте, помогает ли это. Это создаст манекены всех категорий.

categories = {'a', 'b', 'c', 'd', 'e'}

categoriesSeries = pd.Series(list(categories))
pd.get_dummies(categoriesSeries)

Результат

   a  b  c  d  e
0  0  0  0  0  1
1  0  0  1  0  0
2  0  0  0  1  0
3  0  1  0  0  0
4  1  0  0  0  0

Теперь, чтобы создать пустышку для одного элемента, вы можете передать индекс следующим образом.

pd.get_dummies(categoriesSeries[4])

Результат

   a
0  1
...