Python生成矩阵

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

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就可以得到关系矩阵了

您好，我是有问必答小助手，您的问题已经有小伙伴解答了，您看下是否解决，可以追评进行沟通哦~

如果有您比较满意的答案 / 帮您提供解决思路的答案，可以点击【采纳】按钮，给回答的小伙伴一些鼓励哦～～

ps:问答VIP仅需29元，即可享受5次/月 有问必答服务，了解详情>>>https://vip.csdn.net/askvip?utm_source=1146287632