Narysuj inny kolor dla węzłów w networkx na podstawie ich wartości węzła

Question

Mam duży wykres węzłów i skierowanych krawędzi. Ponadto mam dodatkową listę wartości przypisanych do każdego węzła.

Chcę teraz zmienić kolor każdego węzła zgodnie z jego wartością węzła. A więc np. Węzły rysunkowe o bardzo wysokiej wartości czerwonej i te o niskiej wartości niebieskiego (podobne do mapy termicznej). Czy jest to w jakiś sposób łatwe do osiągnięcia? Jeśli nie z networkx, jestem również otwarty na inne biblioteki w Pythonie.

Answer 1

import networkx as nx 
import numpy as np 
import matplotlib.pyplot as plt 

G = nx.Graph() 
G.add_edges_from(
    [('A', 'B'), ('A', 'C'), ('D', 'B'), ('E', 'C'), ('E', 'F'), 
    ('B', 'H'), ('B', 'G'), ('B', 'F'), ('C', 'G')]) 

val_map = {'A': 1.0, 
      'D': 0.5714285714285714, 
      'H': 0.0} 

values = [val_map.get(node, 0.25) for node in G.nodes()] 

nx.draw(G, cmap=plt.get_cmap('jet'), node_color=values) 
plt.show() 

daje

---

Numery values są połączone z węzłami G.nodes(). Oznacza to, że pierwsza liczba w values jest związana z pierwszym węzłem w G.nodes() i podobnie dla drugiego, i tak dalej.