python - 如何在 Python 中创建共现矩阵？

Question

我有一个 N 的数据框列。数据框中的每个元素都在 0 范围内, N-1 .
例如，我的数据帧可能类似于( N=3 ):

    A   B   C
0   0   2   0
1   1   0   1
2   2   2   0
3   2   0   0
4   0   0   0

我想创建一个大小为 N x N 的共现矩阵(如果有不同的标准名称，请纠正我)，其中每个元素 ij 包含元素 i 和 j 假定相同值的次数。

    A   B   C
A   x   2   3
B   2   x   2
C   3   2   x

其中，例如， matrix[0, 1]意味着 A 和 B 假定相同的值 2 次。
我不在乎对角线上的值。
最聪明的方法是什么？

Answer 1

DataFrame.corr
我们可以定义一个自定义的可调用函数来计算数据帧列之间的相关性，这个可调用函数将两个一维 numpy 数组作为其输入参数，并返回这两个数组中元素彼此相等的次数的计数

df.corr(method=lambda x, y: (x==y).sum())

     A    B    C
A  1.0  2.0  3.0
B  2.0  1.0  2.0
C  3.0  2.0  1.0