怎么给networkx绘制的网络图,按节点颜色添加图例,展示代码
时间: 2024-05-10 10:14:56 浏览: 223
怎么用软件画网络图.docx
你可以使用 matplotlib 中的 `legend` 函数来添加图例。以下是一个使用 networkx 绘制网络图并按节点颜色添加图例的示例代码:
```python
import networkx as nx
import matplotlib.pyplot as plt
# 创建一个简单的网络图
G = nx.Graph()
G.add_edges_from([(1,2),(1,3),(2,3),(2,4),(3,4),(4,5),(4,6),(5,6)])
# 按节点的度数生成节点颜色
node_colors = [G.degree(n) for n in G.nodes()]
# 绘制网络图
pos = nx.spring_layout(G)
nx.draw_networkx_nodes(G, pos, node_color=node_colors, cmap=plt.cm.Blues)
nx.draw_networkx_edges(G, pos)
nx.draw_networkx_labels(G, pos)
# 添加图例
plt.colorbar()
# 显示图形
plt.show()
```
运行该代码,将会生成一个按节点颜色添加图例的网络图。
阅读全文