Блог

У меня есть график, и после некоторой классификации я извлек вектор членства следующим образом:

 vector membership is : [1, 1, 1, 1, 1, 2, 1, 2, 2, 2, 3, 3, 3, 3]    
  

это означает, что номер его узла 0:4 находится в том же классе.
Теперь мне нужно нарисовать график, чтобы узлы с одинаковыми номерами имели одинаковую метку. Количество меток здесь равно 3.

Какой самый простой способ сделать это?

Сначала вам нужно создать график, в котором у вас есть атрибут, содержащий метку для этого узла

 import matplotlib.pyplot as plt
import networkx as nx

member_labels = [1, 1, 1, 1, 1, 2, 1, 2, 2, 2, 3, 3, 3, 3]
node_ids = list(range(len(member_labels)))

node_label_dict = dict(zip(node_ids, member_labels))

G = nx.Graph()
G.add_nodes_from(node_ids)
nx.set_node_attributes(G, node_label_dict, 'label')
  

Итак, теперь ваш график имеет следующие узлы и атрибуты

 print(G.nodes(data=True))
#NodeDataView({0: {'label': 1}, 1: {'label': 1}, 2: {'label': 1}, 3: {'label': 1}, 4: {'label': 1}, 5: {'label': 2}, 6: {'label': 1}, 7: {'label': 2}, 8: {'label': 2}, 9: {'label': 2}, 10: {'label': 3}, 11: {'label': 3}, 12: {'label': 3}, 13: {'label': 3}})
  

Теперь вам просто нужно нарисовать узлы и метки

 pos=nx.nx.spring_layout(G)
nx.nx.draw_networkx_nodes(G, pos)
  

Обновить

 # Fixed the variable name
labels = nx.nx.draw_networkx_labels(G,pos, labels=node_label_dict)