Vẽ màu khác nhau cho các nút trong mạngx dựa trên giá trị nút của chúng

Question

Tôi có một biểu đồ lớn các nút và các cạnh được hướng. Hơn nữa, tôi có một danh sách bổ sung các giá trị được gán cho mỗi nút.

Bây giờ tôi muốn thay đổi màu của mỗi nút theo giá trị nút của chúng. Ví dụ: vẽ các nút có giá trị rất cao màu đỏ và các nút có giá trị xanh dương thấp (tương tự như bản đồ nhiệt). Đây có phải là cách dễ dàng nhất có thể đạt được không? Nếu không với networkx, tôi cũng mở cho các thư viện khác bằng Python.

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() 

mang

---

Các con số trong values được kết hợp với các nút trong G.nodes(). Tức là, số đầu tiên trong số values được kết hợp với nút đầu tiên trong số G.nodes() và tương tự cho số thứ hai, v.v.