将 Pandas Dataframe 列转换为一个热标签

Question

我有一个类似这样的熊猫数据框：

  Col1   ABC
0  XYZ    A
1  XYZ    B
2  XYZ    C

通过在 ABC 列上使用 pandas 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 函数，然后将所有列合并到我想要的列中。我找到了很多解释如何将多列组合为字符串的答案，例如：Combine two columns of text in dataframe in pandas/python。但我想不出一种方法将它们组合成一个列表。

这个问题介绍了使用sklearn的OneHotEncoder的想法，但我无法让它工作。 How do I one-hot encode one column of a pandas dataframe?

还有一件事：我遇到的所有答案都有解决方案，其中列名必须在组合它们时手动输入。有没有办法使用Dataframe.iloc()或拼接机制将列组合成一个列表？

Answer 1

这里是一个使用sklearn.preprocessing.LabelBinarizer的例子：

In [361]: from sklearn.preprocessing import LabelBinarizer

In [362]: lb = LabelBinarizer()

In [363]: df['new'] = lb.fit_transform(df['ABC']).tolist()

In [364]: df
Out[364]:
  Col1 ABC        new
0  XYZ   A  [1, 0, 0]
1  XYZ   B  [0, 1, 0]
2  XYZ   C  [0, 0, 1]

熊猫替代品：

In [370]: df['new'] = df['ABC'].str.get_dummies().values.tolist()

In [371]: df
Out[371]:
  Col1 ABC        new
0  XYZ   A  [1, 0, 0]
1  XYZ   B  [0, 1, 0]
2  XYZ   C  [0, 0, 1]

Answer 2

你可以使用tolist():

df['ABC'] = pd.get_dummies(df.ABC).values.tolist()

  Col1        ABC
0  XYZ  [1, 0, 0]
1  XYZ  [0, 1, 0]
2  XYZ  [0, 0, 1]

Answer 3

如果你有这样的 pd.DataFrame：

>>> df
  Col1  A  B  C
0  XYZ  1  0  0
1  XYZ  0  1  0
2  XYZ  0  0  1

你总是可以这样做：

>>> df.apply(lambda s: list(s[1:]), axis=1)
0    [1, 0, 0]
1    [0, 1, 0]
2    [0, 0, 1]
dtype: object

注意，这本质上是行上的 for 循环。请注意，列不具有list 数据类型，它们必须是object，这将使您的数据帧操作无法利用numpy 的速度优势。

Answer 4

如果您有一个带有分类列 ABC 的数据框 df，那么您可以使用它来创建一个新的 one-hot 向量列

df['new_column'] = list(pandas.get_dummies(df['AB]).get_values())