nx.draw_networkx_nodes(G, pos, nodelist=list(G), node_color=partition.values(), cmap='viridis',node_size=30)增加lable属性
时间: 2024-12-16 10:18:33 浏览: 6
在`nx.draw_networkx_nodes`函数中增加`label`属性,你可以通过将`label`参数传递给函数,为每个节点添加标签。假设`partition`是一个字典,其中的键是节点,值是你想要作为标签显示的信息。以下是修改后的示例代码:
```python
import networkx as nx
import matplotlib.pyplot as plt
from matplotlib import cm
# 假设G是一个已经构建好的NetworkX图,pos是节点位置字典
# partition是一个包含节点分类的字典,例如 {node: category}
# 使用partition的值作为节点颜色,并添加标签
node_labels = {node: f"{node}: {partition[node]}" for node in nodelist} # 将partition的值转换为标签格式
node_colors = [partition[node] for node in nodelist]
fig, ax = plt.subplots()
nx.draw_networkx_nodes(G, pos, nodelist=nodelist, node_color=node_colors, cmap=cm.viridis, node_size=30)
nx.draw_networkx_labels(G, pos, labels=node_labels, font_weight='bold') # 添加节点标签
plt.show()
```
在这个例子中,我们首先根据`partition`字典生成了每个节点的标签(将节点名称与对应的类别结合),然后在`nx.draw_networkx_labels`部分将这些标签添加到图上,使用`font_weight='bold'`使标签更醒目。
阅读全文