node_shape参数
时间: 2023-07-08 21:31:20 浏览: 385
`node_shape`是用于指定节点形状的参数,在`nx.draw`和`nx.draw_networkx`函数中都可以使用。它可以是一个节点形状字符串(如`'o'`表示圆形、`'s'`表示正方形、`'^'`表示三角形等),也可以是一个自定义的节点形状函数或类。
一些常见的节点形状字符串及其含义如下:
- `'o'`:圆形节点
- `'s'`:正方形节点
- `'d'`:菱形节点
- `'^'`:向上的三角形节点
- `'v'`:向下的三角形节点
- `'>'`:向右的三角形节点
- `'<'`:向左的三角形节点
- `'p'`:五边形节点
- `'h'`:六边形节点
下面是一个使用`node_shape`参数的示例代码:
```python
import networkx as nx
import matplotlib.pyplot as plt
# 创建一个无向图
G = nx.Graph()
# 添加一些节点和边
G.add_nodes_from(['A', 'B', 'C', 'D'])
G.add_edges_from([('A', 'B'), ('A', 'C'), ('B', 'C'), ('B', 'D'), ('C', 'D')])
# 生成节点的布局
pos = nx.spring_layout(G)
# 指定节点的形状
node_shapes = ['o', 's', '^', 'd']
# 绘制图形
nx.draw(G, pos=pos, with_labels=True, node_shape=node_shapes)
# 显示图形
plt.show()
```
在上面的代码中,我们首先创建了一个无向图`G`,然后添加了一些节点和边。接着,我们使用`nx.spring_layout`函数生成节点的布局,并将结果保存在`pos`变量中。最后,我们使用`node_shapes`列表指定了每个节点的形状,并将其作为参数传递给`nx.draw`函数,绘制出了带有不同形状节点的图形。