Networkx：来自熊猫数据框的网络图答案

【问题标题】：Networkx: Network graph from pandas dataframeNetworkx：来自熊猫数据框的网络图
【发布时间】：2021-05-10 04:08:17
【问题描述】：

我有以下数据集：

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

数据框显示了不同投资公司在一年内投资同一家公司的时间。我只想创建 Firm_ID 之间的连接的网络图。例如，Ampersand 和 BancBoston 都投资了同一家公司，因此应该建立联系。我试过的代码是：

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

生成如下图：

这显示了 Company_ID 和 Firm_Name 的连接。我只想将这些公司作为节点，如果他们投资了同一家公司，它们就会连接起来。我没有发现任何使用 networkx 的类似问题或类似数据集。非常感谢任何帮助！

【问题讨论】：

标签： python pandas networkx

【解决方案1】：

试试merge

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

【讨论】：

干杯！这似乎做到了！