Окрашивание определенных узлов в сети x

Question

У меня есть график в сетиx, который я хочу построить.Все работает нормально, но я решил, что было бы полезно покрасить определенные узлы.Как мне это сделать?

Я видел, что это можно сделать с помощью команды

draw_networkx_nodes(G, pos, nodelist=None, node_size=300, node_color='r', node_shape='o', alpha=1.0, cmap=None, vmin=None, vmax=None, ax=None, linewidths=None, label=None, **kwds)

, установив для ключевого слова color и cmap правильное значение.Но все примеры, которые я видел до сих пор, используют cmap=plt.get_cmap(some string), которая является стандартной функцией, которая не кажется мне полезной.

Чтобы уточнить: у меня есть несколько списков узлов, и я хочу покрасить все узлы всписок в том же цвете

Answer 1

Вы можете нарисовать ваши узлы в группах на основе цвета, аналогичного примеру , который они предоставляют.Для каждой нарисованной группы просто укажите node_color для нужного цвета.

import networkx as nx

G = nx.cubical_graph()
pos = nx.spring_layout(G)

nodes = {
    'r': [1, 3, 5],
    'b': [0, 2],
    'g': [4]
}
for node_color, nodelist in nodes.items():
    nx.draw_networkx_nodes(G, pos, nodelist=nodelist, node_color=node_color)

labels = {x: x for x in G.nodes}
nx.draw_networkx_labels(G, pos, labels, font_size=16, font_color='w')