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

Question

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

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

但我得到了

  

DeprecationWarning：传递1d数组作为数据在0.17中弃用   并在0.19中提出ValueError。

有解决方法吗？

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