Networkx:数据帧中的网络图

Question

我有以下数据集：

Company_ID  Firm_Name
125911      Ampersand 
125911      BancBoston 
32679       BP Corp 
74240       CORNING 
32679       DIEBOLD 
32679       DIEBOLD 
74240       Fidelity 
74240       Greylock
32679       INCO 
67734       INCO 
67734       Innova
32679       Kleiner 
67734       Kleiner 
67734       Kleiner 
67734       Mayfield
32679       Pliant 
67734       Pliant 
67734       Sofinnova 
43805       Warburg 

数据框显示了不同的投资公司在一年内投资同一家公司的时间。我只想创建公司ID之间连接的网络图。例如，安培生和班克博斯顿均投资于同一家公司，因此应相互关联。我尝试过的代码是：

G = nx.Graph()
G = nx.from_pandas_edgelist(df, 'Company_ID', 'Firm_Name')
nx.draw_shell(H, with_labels=True)

这将生成以下图形：

这显示了公司ID和公司名称的连接。我只想把这些公司作为节点，如果它们投资于同一家公司，它们就在这里相互连接。在使用networkx的地方，我没有发现任何类似的问题或类似的数据集。非常感谢您的帮助

Answer 1

尝试使用merge

out = df.merge(df,on=['Company_ID'])
G = nx.Graph()
G = nx.from_pandas_edgelist(df, 'Firm_Name_x', 'Firm_Name_y')