python - Networkx 图形绘制节点权重

Question

我想为无向图中的每个节点分配节点权重。我使用以下 MWE:

import sys
import matplotlib.pyplot as plt
import networkx as nx
G = nx.Graph()
G.add_node(0)
G.add_node(1, weight=2)
G.add_node(2, weight=3)
nx.draw(G, with_labels=True)
plt.show()

然后我有一个如下形式的图:

我想绘制一个图表，其中节点旁边的权重以新颜色给出，例如:

实现这一点的最简单方法是什么？在 SO 上， Material 主要用于边缘权重，或改变节点大小。节点权重。

Answer 1

您可以将 labels 属性与相应的字典一起使用，将 node_color 属性与相应的列表一起使用。对于此代码:

G = nx.Graph()
G.add_node(0, weight=8)
G.add_node(1, weight=5)
G.add_node(2, weight=3)
labels = {n: G.nodes[n]['weight'] for n in G.nodes}
colors = [G.nodes[n]['weight'] for n in G.nodes]
nx.draw(G, with_labels=True, labels=labels, node_color=colors)

Networkx 将绘制:

如果你想同时绘制节点ID和权重，你可以这样写:

标签 = {n: str(n) + '; ' + str(G.nodes[n]['weight']) for G.nodes 中的 n}

<小时/>

如果节点中缺少weight属性并且想要绘制它们，可以使用以下代码:

labels = {
    n: str(n) + '\nweight=' + str(G.nodes[n]['weight']) if 'weight' in G.nodes[n] else str(n)
    for n in G.nodes
}

我认为在附近具有不同颜色的节点绘制权重几乎是不可能的。这是我能给你的最好建议。