如何优雅地对熊猫中的一系列列表进行热编码

Question

所以我有以下数据：

>>> test = pd.Series([['a', 'b', 'e'], ['c', 'a'], ['d'], ['d'], ['e']])
>>> test

0    [a, b, e]
1       [c, a]
2          [d]
3          [d]
4          [e]

我正在尝试将列表中的所有数据单热编码回我的数据帧。看起来像这样：

>>> pd.DataFrame([[1, 1, 0, 0, 1], [1, 0, 1, 0, 0],
              [0, 0, 0, 1, 0], [0, 0, 0, 1, 0],
              [0, 0, 0, 0, 1]],
             columns = ['a', 'b', 'c', 'd', 'e'])

    a   b   c   d   e
0   1   1   0   0   1
1   1   0   1   0   0
2   0   0   0   1   0
3   0   0   0   1   0
4   0   0   0   0   1

我试过研究，我发现了类似的问题，但没有像这样的。我尝试过：

test.apply(pd.Series)

但这并不能完全实现单热方面。这只是以任意顺序打开我的列表。我确信我可以找到一个冗长的解决方案，但我很高兴听到是否有更优雅的方式来执行此操作。

谢谢！

编辑：我知道我可以遍历我的test系列，然后为找到的每个唯一值创建一个列，然后返回并test再次迭代，将所述列标记为唯一值。但这对我来说似乎不是很讨人喜欢，我相信有一种更优雅的方式来做到这一点。

Answer 1

MultiLabelBinarizersklearn对于这些问题，来自库的效率更高。应该优先于applywith pd.Series。这是一个演示：

import pandas as pd
from sklearn.preprocessing import MultiLabelBinarizer

test = pd.Series([['a', 'b', 'e'], ['c', 'a'], ['d'], ['d'], ['e']])

mlb = MultiLabelBinarizer()

res = pd.DataFrame(mlb.fit_transform(test),
                   columns=mlb.classes_,
                   index=test.index)

结果

   a  b  c  d  e
0  1  1  0  0  1
1  1  0  1  0  0
2  0  0  0  1  0
3  0  0  0  1  0
4  0  0  0  0  1