用Python画一个树状图
时间: 2023-11-16 22:03:29 浏览: 43
好的,以下是一个简单的Python代码示例,可以用来画一个树状图:
```python
import matplotlib.pyplot as plt
# 定义节点坐标和标签
nodes_pos = {
0: (0.5, 0.9, 'A'),
1: (0.2, 0.6, 'B'),
2: (0.8, 0.6, 'C'),
3: (0.1, 0.3, 'D'),
4: (0.4, 0.3, 'E'),
5: (0.7, 0.3, 'F'),
6: (0.9, 0.3, 'G')
}
# 定义节点间的连接关系
edges = [
(0, 1),
(0, 2),
(1, 3),
(1, 4),
(2, 5),
(2, 6)
]
# 绘制树状图
fig = plt.figure(figsize=(6, 6))
ax = fig.add_subplot(111)
for edge in edges:
start = nodes_pos[edge[0]]
end = nodes_pos[edge[1]]
ax.plot([start[0], end[0]], [start[1], end[1]], 'k-')
for node in nodes_pos:
pos = nodes_pos[node]
ax.text(pos[0], pos[1], pos[2], ha='center', va='center', fontsize=14)
ax.axis('off')
plt.show()
```
运行以上代码,可以得到一个简单的树状图:
```
A
/ \
B C
/ \ / \
D E F
|
G
```