Pandas: Wie konvertiert man einen One-hot-codierten Datenrahmen in eine Adjazenzmatrix?

Question

Ich habe einen Datenrahmen df, der Form:

 Col1 Col2 Col3 
0  0  1  0 
1  1  1  0 
2  0  1  1 
3  1  1  0 

Ich brauche eine neue df der Form:

 Col1 Col2 Col3 
Col1  0  2  0  
Col2  2  0  1 
Col3  0  1  0 

Grundsätzlich sind die Werte, die die Kookkurrenzen zweier darstellen angegebene Spalten für alle Zeilen.

Wie gehe ich vor?

Answer 1

einfach matrix-multiplication dort nutzen -

In [21]: df_out = df.T.dot(df) 

In [22]: np.fill_diagonal(df_out.values, 0) 

In [23]: df_out 
Out[23]: 
     Col1 Col2 Col3 
Col1  0  2  0 
Col2  2  0  1 
Col3  0  1  0