python – 如何对一个pandas数据帧的一列进行单热编码？

我正在尝试对数据帧的一列进行单热编码.

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

但我得到了

DeprecationWarning: Passing 1d arrays as data is deprecated in 0.17
and willraise ValueError in 0.19.

这有解决方法吗？

解决方法:

我想你可以使用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

编辑：

如果输入是带列表的列,首先转换为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