python数据可视化关系图代码
时间: 2023-10-12 20:11:48 浏览: 67
Python数据可视化课后习题_答案.docx
以下是使用 NetworkX 和 Matplotlib 绘制简单关系图的 Python 代码示例:
```python
import networkx as nx
import matplotlib.pyplot as plt
# 创建一个简单的关系图
G = nx.Graph()
G.add_edges_from([(1, 2), (1, 3), (2, 3), (3, 4), (4, 5), (4, 6), (5, 6)])
# 设置节点和边的样式
node_color = ['r', 'g', 'b', 'y', 'c', 'm']
edge_color = ['r', 'g', 'b', 'y', 'c', 'm']
node_size = [800, 600, 400, 200, 100, 50]
edge_width = [2, 3, 4, 5, 6, 7]
# 绘制关系图
pos = nx.spring_layout(G)
nx.draw_networkx_nodes(G, pos, node_color=node_color, node_size=node_size)
nx.draw_networkx_edges(G, pos, edge_color=edge_color, width=edge_width)
# 显示关系图
plt.axis('off')
plt.show()
```
这段代码会绘制一个包含 6 个节点和 7 条边的简单关系图,并设置节点和边的样式,最后显示关系图。你可以根据自己的需求修改节点和边的样式,或者使用其他库进行绘制。
阅读全文