如何单速编码PANDAS数据框架的一列

Question

我正在尝试使用一个hot编码数据框的一列。

enc = OneHotEncoder()
minitable = enc.fit_transform(df["ids"])

但是我得到了

不弃用战争：随着数据弃用为0.17，传递1D阵列 和willaise valueerror在0.19中。

是否有解决方法？

Answer 1

我认为您可以使用get_dummies：

df = pd.DataFrame({'ids':['a','b','c']})
print (df)
  ids
0   a
1   b
2   c
print (df.ids.str.get_dummies())
   a  b  c
0  1  0  0
1  0  1  0
2  0  0  1

编辑：

如果输入是用lists列，则首先铸造为str，请通过strip删除[]并致电get_dummies：

df = pd.DataFrame({'ids':[[0,4,5],[4,7,8],[5,1,2]]})
print(df)
         ids
0  [0, 4, 5]
1  [4, 7, 8]
2  [5, 1, 2]
print (df.ids.astype(str).str.strip('[]').str.get_dummies(', '))
   0  1  2  4  5  7  8
0  1  0  0  1  1  0  0
1  0  0  0  1  0  1  1
2  0  1  1  0  1  0  0