Python生成矩阵

Question

求大佬指点 做公司的连锁董事网络，一个公司对应多个董事，如果两家公司有相同的董事则设两家公司的关系为1，否则为0。需要生成公司与公司之间的关系矩阵。这个要怎么弄？

Answer 1

import pandas as pd
import numpy as np
company = {'company1':['name1','name2','name3'],'company2':['name3','name4'],'company3':['name4','name5']}
data = np.zeros((len(company),len(company)))
df=pd.DataFrame(data,columns=company.keys(),index=company.keys())
for i in company.keys():
    for j in company.keys():
        x = [x for x in company[i] if x in company[j]]
        if len(x)>0:
            df[i][j] = 1
            df[j][i] = 1

这里需要把公司和董事写成一个字典，最后打印df就可以得到关系矩阵了