Простая раскраска узлов в NetworkX - PullRequest
0 голосов
/ 06 февраля 2020

У меня есть следующий график:

import networkx as nx
import matplotlib.pyplot as plt
G = nx.Graph()
G.add_edge('V1', 'R1')
G.add_edge('R1', 'R2')
G.add_edge('R2', 'R3')
G.add_edge('R2', 'R4')
G.add_edge('R3', 'Z')
G.add_edge('R4', 'Z')
nx.draw(G, with_labels=True)
colors = ['yellow' if node.starswith('V') else 'red' if node.startswith('R', else 'black')]
plt.show()

Как бы я раскрасил различные узлы, как показано выше?

1 Ответ

1 голос
/ 06 февраля 2020

Вам необходимо передать colors как атрибут node_color в функции nx.draw. Вот код,

import networkx as nx
import matplotlib.pyplot as plt
G = nx.Graph()
G.add_edge('V1', 'R1')
G.add_edge('R1', 'R2')
G.add_edge('R2', 'R3')
G.add_edge('R2', 'R4')
G.add_edge('R3', 'Z')
G.add_edge('R4', 'Z')

# I have added a function which returns
# a color based on the node name
def get_color(node):
  if node.startswith('V'):
    return 'yellow'
  elif node.startswith('R'):
    return 'red'
  else:
    return 'black'

colors = [ get_color(node) for node in G.nodes()]
# ['yellow', 'red', 'red', 'red', 'red', 'black']


# Now simply pass this `colors` list to `node_color` attribute 
nx.draw(G, with_labels=True, node_color=colors)
plt.show()

Node Colors

Вот ссылка на рабочий код в Google Colab Notebook .

Ссылки:

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...