将Pandas Dataframe列转换为一个热标签(Converting a Pandas Dataframe column into one hot labels)

 我有一个与此类似的pandas数据帧： 
 
  Col1   ABC
0  XYZ    A
1  XYZ    B
2  XYZ    C
 
 通过在列ABC上使用熊猫get_dummies()函数，我可以得到这个： 
 
  Col1   A   B   C
0  XYZ   1   0   0
1  XYZ   0   1   0
2  XYZ   0   0   1
 
 虽然我需要这样的东西，ABC列有一个list / array数据类型： 
 
  Col1    ABC
0  XYZ    [1,0,0]
1  XYZ    [0,1,0]
2  XYZ    [0,0,1]
 
 我尝试使用get_dummies函数，然后将所有列组合到我想要的列中。 我发现很多答案解释了如何将多个列组合成字符串，如下所示： 在pandas / python中的数据框中组合两列文本 。 但我无法想出将它们组合成列表的方式。 
 
 这个问题介绍了使用sklearn的OneHotEncoder的想法，但我无法让它工作。 如何对一个pandas数据帧的一列进行单热编码？ 
 
 还有一件事：我遇到的所有答案都有解决方案，其中列名称必须在组合时手动输入。 有没有办法使用Dataframe.iloc()或拼接机制将列组合成一个列表？ 

I have a pandas dataframe similar to this:
 
  Col1   ABC
0  XYZ    A
1  XYZ    B
2  XYZ    C
 
By using the pandas get_dummies() function on column ABC, I can get this:
 
  Col1   A   B   C
0  XYZ   1   0   0
1  XYZ   0   1   0
2  XYZ   0   0   1
 
While I need something like this, where the ABC column has a list / array datatype:
 
  Col1    ABC
0  XYZ    [1,0,0]
1  XYZ    [0,1,0]
2  XYZ    [0,0,1]
 
I tried using the get_dummies function and then combining all the columns into the column which I wanted. I found lot of answers explaining how to combine multiple columns as strings, like this: Combine two columns of text in dataframe in pandas/python. But I cannot figure out a way to combine them as a list.
 
This question introduced the idea of using sklearn's OneHotEncoder, but I couldn't get it to work. How do I one-hot encode one column of a pandas dataframe?
 
One more thing: All the answers I came across had solutions where the column names had to be manually typed while combining them. Is there a way to use Dataframe.iloc() or splicing mechanism to combine columns into a list?

原文：https://stackoverflow.com/questions/47127388