networkx通过深度优先遍历找出某个点旁边的所有点

最新推荐文章于 2024-08-26 13:28:57 发布

原创最新推荐文章于 2024-08-26 13:28:57 发布 · 402 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#深度优先 #算法

networkx 专栏收录该内容

4 篇文章

订阅专栏

本文展示了如何使用Python的networkx库创建一个图，通过pandas处理数据，然后利用matplotlib进行图形化展示，以及执行深度优先搜索。代码实例包括数据导入、图的构建和特定节点的搜索操作。

import networkx as nx
import pandas as pd
import matplotlib.pyplot as plt

source = [1, 2, 3, 4, 5]
target = [2, 3, 4, 5, 6]
id = [10, 11, 12, 13, 14]
df = pd.DataFrame({'id': id, 'source': source, 'target': target})
print(df)

graph = nx.Graph()

pipe_data_dict = df.to_dict(orient='records')
for pipe in pipe_data_dict:
    graph.add_edge(pipe['source'], pipe['target'], gid=pipe['id'])

# 保存图
plt.figure(figsize=(6, 4))
pos = nx.circular_layout(graph)
nx.draw(graph, pos=pos, with_labels=True)
plt.savefig("demo", dpi=1000, bbox_inches='tight')

for source, target in nx.dfs_edges(graph, 2, depth_limit=1):
    print("source:", source, "target:", target)
    print("id:", graph[source][target]['gid'])

在这里插入图片描述

结果

   id  source  target
0  10       1       2
1  11       2       3
2  12       3       4
3  13       4       5
4  14       5       6


source: 2 target: 1
id: 10
source: 2 target: 3
id: 11