时间戳

首页 > 解决方案 > 如何用列表中的单词制作二进制熊猫矩阵？

python - 如何用列表中的单词制作二进制熊猫矩阵？

问题描述

我有一个数据框，其中有一列包含每一行的列表：

df['list1'][0] = ['a','b','c','d']

df['list1'][1] = [,'b','c','d', 'e']

现在我想从中创建一个二进制矩阵，将字母作为列名，如果该行上的列表包含字母，则为 1，如果不是 0。我找不到解决方案。有人可以帮我吗？

标签： pythonpandasmatrix

解决方案

如果性能很重要，请使用MultiLabelBinarizer：

df = pd.DataFrame({'list1':[ ['a','b','c','d'], ['b','c','d', 'e']]})
print (df)
          list1
0  [a, b, c, d]
1  [b, c, d, e]

from sklearn.preprocessing import MultiLabelBinarizer

mlb = MultiLabelBinarizer()
df1 = pd.DataFrame(mlb.fit_transform(df['list1']),columns=mlb.classes_)

或get_dummies与DataFrame构造函数：

df1 = (pd.get_dummies(pd.DataFrame(df['list1'].values.tolist()), prefix_sep='', prefix='')
         .max(level=0, axis=1))

另一种解决方案，但如果使用大数据会很Series.str.join慢Series.str.get_dummies：

df1 = df['list1'].str.join('|').str.get_dummies()

print (df1)
   a  b  c  d  e
0  1  1  1  1  0
1  0  1  1  1  1

编辑：如果需要删除list1列的原始列DataFrame.pop用于提取列DataFrame.join：

df = pd.DataFrame({'list1':[ ['a','b','c','d'], ['b','c','d', 'e']],
                   'col1':[1,2],
                   'col2':list('XY')})
print (df)
          list1  col1 col2
0  [a, b, c, d]     1    X
1  [b, c, d, e]     2    Y

from sklearn.preprocessing import MultiLabelBinarizer

mlb = MultiLabelBinarizer()
df1 = pd.DataFrame(mlb.fit_transform(df.pop('list1')),columns=mlb.classes_, index=df.index)
print (df1)
   a  b  c  d  e
0  1  1  1  1  0
1  0  1  1  1  1

df1 = (pd.get_dummies(pd.DataFrame(df.pop('list1').values.tolist()), prefix_sep='', prefix='')
         .max(level=0, axis=1))

df1 = df.pop('list1').str.join('|').str.get_dummies()

df = df.join(df1)
print (df)
   col1 col2  a  b  c  d  e
0     1    X  1  1  1  1  0
1     2    Y  0  1  1  1  1

推荐阅读