Networkx Graph 绘图节点权重

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 上，材料主要用于边缘权重，或者改变节点大小 w.r.t。节点权重。

Answer 1

您可以将labels 属性与相应的dict 和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 和它的权重，你可以这样写：

labels = {n: str(n) + '; ' + str(G.nodes[n]['weight']) for n in G.nodes}

---

如果您在节点中缺少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
}

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