Превращение матрицы в фиктивную матрицу

Question

Я сгенерировал список комбинаций и хотел бы превратить его в матрицу «пустышек»

 import pandas as pd
 from itertools import combinations
 comb = pd.DataFrame(list(combinations(range(1, 6), 4)))

    0 1 2 3
 0  1 2 3 4
 1  1 2 3 5
 2  1 2 4 5
 3  1 3 4 5
 4  2 3 4 5

хотел бы превратить указанный выше кадр данных в такой, как показано ниже.Благодарю.

    1 2 3 4 5
 0  1 1 1 1 0
 1  1 1 1 0 1
 2  1 1 0 1 1
 3  1 0 1 1 1
 4  0 1 1 1 1 

Answer 1

Вы можете использовать MultiLabelBinarizer:

from sklearn.preprocessing import MultiLabelBinarizer

lb = MultiLabelBinarizer()
df = pd.DataFrame(lb.fit_transform(comb.values), columns= lb.classes_)

print (df)

   1  2  3  4  5
0  1  1  1  1  0
1  1  1  1  0  1
2  1  1  0  1  1
3  1  0  1  1  1
4  0  1  1  1  1