python 共现_Python单词列表中的共现矩阵

I have a list of names like:

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

and a list of documents, that in each documents some of these names are mentioned.

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

I would like to get an output as a matrix of co-occurrences like:

A B C D

A 0 2 1 1

B 2 0 2 1

C 1 2 0 1

D 1 1 1 0

There is a solution (Creating co-occurrence matrix) for this problem in R, but I couldn't do it in Python. I am thinking of doing it in Pandas, but yet no progress!

解决方案

Obviously this can be extended for your purposes, but it performs the general operation in mind:

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)