如何获得一个值的熊猫系列虚拟表示

Question

假设我有 5 个不同的类别

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

是否可以使用 Pandas 的get_dummies 来获得一个元素的虚拟表示？即，假设我有

element = 'a'

变成

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

Answer 1

请检查这是否有帮助。这将生成所有类别的假人。

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