嵌套单词列表中的共现矩阵

Question

我有一个类似的名字列表：

names = ['A', 'B', 'C', 'D']

以及文档列表，在每个文档中都提到了其中一些名称。

document =[['A', 'B'], ['C', 'B', 'K'],['A', 'B', 'C', 'D', 'Z']]

我想获得输出作为共现矩阵，例如：

  A  B  C  D
A 0  2  1  1
B 2  0  2  1
C 1  2  0  1
D 1  1  1  0

R中有一个解决此问题的方法（创建共现矩阵），但我无法在Python中做到这一点。我正在考虑在Pandas进行此操作，但没有任何进展！

Answer 1

显然，可以根据您的目的进行扩展，但是它会执行以下常规操作：

import math

for a in 'ABCD':
    for b in 'ABCD':
        count = 0

        for x in document:
            if a != b:
                if a in x and b in x:
                    count += 1

            else:
                n = x.count(a)
                if n >= 2:
                    count += math.factorial(n)/math.factorial(n - 2)/2

        print '{} x {} = {}'.format(a, b, count)