我有以下数据集：
firm_id_1 firm_id_2
1 2
1 4
1 5
2 1
2 3
3 2
3 6
4 1
4 5
4 6
5 4
5 7
6 3 ....
我想画图的网络firm_id = 1。换句话说，我想看到一个图表，该图表显示firm_id = 1直接通过公司2连接到2、4、5，并通过公司2间接连接到3，通过公司4连接到6，通过公司5间接连接到7。我绘制了从开始到每个节点（firm_id）的最短距离firm_id=1。我的数据中有3000个节点，我知道公司1到达少于9个顶点的所有节点。如何在Python中绘制图形？
我将从一个名为NetworkX的库开始。我不确定我是否了解您要查找的所有内容，但是我认为这应该足够接近，您可以对其进行修改。
该程序将从文本文件graphdata.txt中加载数据，并用空格分隔，然后将其作为边缘添加。
然后它将计算从1开始到所有节点的最短路径，如果距离大于9，则进行打印。有关更多详细信息，请参阅文档。
最后，它将使用弹性布局将图形呈现到名为mynetwork.png的文件并呈现到屏幕。
3000个节点可能需要/可能不需要某些优化。
希望这可以帮助！
import networkx as nx
import matplotlib.pyplot as plt
graph = nx.Graph()
with open('graphdata.txt') as f:
for line in f:
firm_id_1, firm_id_2 = line.split()
graph.add_edge(firm_id_1, firm_id_2)
paths_from_1 = nx.shortest_path(graph, "1")
for path in paths_from_1:
if len(paths_from_1[node]) > 9:
print "Shortest path from 1 to", node, "is longer than 9"
pos = nx.spring_layout(graph, iterations=200)
nx.draw(graph, pos)
plt.savefig("mynetwork.png")
plt.show()
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句