使用python编写一个简单的人物关系图
时间: 2024-02-29 07:53:40 浏览: 23
可以使用Python中的NetworkX库来创建和可视化人物关系图。以下是一个简单的示例代码,用于创建和绘制一个类似上面描述的人物关系:
```python
import networkx as nx
import matplotlib.pyplot as plt
# 创建人物关系图
G = nx.Graph()
# 添加人物节点
G.add_nodes_from(['小明', '小红', '小华'])
# 添加边
G.add_edge('小明', '小红', relationship='男女朋友')
G.add_edge('小明', '小华', relationship='好朋友')
G.add_edge('小红', '小华', relationship='同事')
G.add_edge('小华', '小明', relationship='大学同学')
# 绘制人物关系图
pos = nx.spring_layout(G, seed=42)
nx.draw_networkx_nodes(G, pos)
nx.draw_networkx_edges(G, pos)
nx.draw_networkx_labels(G, pos)
edge_labels = nx.get_edge_attributes(G, 'relationship')
nx.draw_networkx_edge_labels(G, pos, edge_labels=edge_labels)
plt.show()
```
运行后,会得到一个可视化的人物关系图,如下所示:
![人物关系图](https://i.imgur.com/1Uw5RJ5.png)