如何对熊猫数据框的一列进行单热编码？

Question

我试图对数据框的一列进行一次热编码。

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

但是我却越来越

DeprecationWarning：传递一维数组作为数据在0.17 弃用，在0.19 willraise 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