networkx画网络图

NetworkX是一个强大的Python库，用于创建、操作和研究复杂网络。它非常适合于可视化网络图，包括无向图、有向图和加权图等。以下是如何使用networkx来绘制简单网络图的基本步骤：

1. 首先，安装NetworkX库，如果尚未安装，可以使用pip命令：

   pip install networkx

2. 导入所需的模块：

import networkx as nx
import matplotlib.pyplot as plt

3. 创建节点和边的数据结构。例如，用字典表示节点及其属性，列表或元组表示边：

nodes = ['A', 'B', 'C', 'D']
edges = [('A', 'B'), ('B', 'C'), ('C', 'D'), ('D', 'A')]

# 或者更复杂一些，如果有节点属性：
node_data = {'A': {'color': 'red'}, 'B': {'color': 'blue'}}
edge_data = [('A', 'B', {'weight': 1}), ('B', 'C', {'weight': 2})]

4. 使用`nx.Graph()`创建网络对象，并添加节点和边：

G = nx.Graph()
G.add_nodes_from(nodes)
G.add_edges_from(edges)

5. 对于带有属性的边，需要使用`add_edge()`函数并传递额外的键值对作为第三参数：

G.add_edge('A', 'B', weight=1)

6. 可视化网络图，这里使用matplotlib：

pos = nx.spring_layout(G)  # 指定布局算法
nx.draw_networkx_nodes(G, pos, node_color=[node_data[node]['color'] for node in G.nodes], cmap='viridis')
nx.draw_networkx_edges(G, pos, edge_color=G.edges(data=True)['weight'], width=[e['weight']/5 for e in G.edges], alpha=0.8)
plt.show()  # 显示图形