使用 networkx 繪制多部分圖

Question

我有這個圖形結構

graph = thisdict['data']['graph']

dict_edges = graph['edges']

edges = []

for edge in dict_edges:
    edges.append((edge['source']['node_id'], edge['target']['node_id']))

print('Source:\t\t\t     Target:\n')
    
for edge in edges:
    print(str(edge))

print('\n')

dict_nodes = graph['nodes']

nodes = {}

for node in dict_nodes:
    nodes[node['id']] = node['name']

print('Node ID:\t\t    Node Name:\n')

for key, value in nodes.items():
    print("'%s':'%s'" %(key, value))

輸出：

Source:                                Target:

('61697b94f74c92a808641ba3', '61697b95f74c92a808641ba4')
('61697b94f74c92a808641ba3', '61697b96f74c92a808641ba5')
('61697b95f74c92a808641ba4', '61697b96f74c92a808641ba6')
('61697b96f74c92a808641ba6', '61697b97f74c92a808641ba7')
('61697b96f74c92a808641ba5', '61697b97f74c92a808641ba7')
('61697b97f74c92a808641ba7', '61697b98f74c92a808641ba8')
('61697b98f74c92a808641ba8', '61697b98f74c92a808641ba9')


Node ID:            Node Name:

'61697b94f74c92a808641ba3':'S3 connector'
'61697b95f74c92a808641ba4':'loader 1'
'61697b96f74c92a808641ba5':'loader 2'   
'61697b96f74c92a808641ba6':'sampler 1'
'61697b97f74c92a808641ba7':'concator'
'61697b98f74c92a808641ba8':'sampler 2'
'61697b98f74c92a808641ba9':'splitter'

我寫了這段代碼來繪制圖形：

nx_graph = nx.Graph()
plt.figure(figsize=(3,3))

for key, value in nodes.items():
    nx_graph.add_node(key, layer = nodes.values())
    #I need to put every node name in a single layer, So I should have 6 layers

for edge in edges:
    nx_graph.add_edge(*edge)

pos = nx.multipartite_layout(nx_graph, subset_key="layer")

nx.draw(nx_graph, pos, labels=nodes, with_labels=True)

plt.show()

它顯示一個錯誤說： TypeError: unsupported operand type(s) for -: 'dict_values' and 'float'

我需要將每個節點名稱放在一個層中，所以我應該有 6 層。 圖層排序應如下所示：

S3 connector --> loader 1 & loader 2
                 loader 1 will give sampler 1
                 loader 2 & sampler 1 will meet in concator
                 concator will give sampler 2
                 sampler 2 will give splitter

Answer 1

發生錯誤是因為當您在for循環中添加節點時，您傳遞的是layer = nodes.values()而不僅僅是layer=value 。 但是，糾正它仍然不會為您提供所需的布局，因為您實際上必須以某種方式指定圖層。 根據您對 6 層的描述，我將它們添加為來自單獨字典的屬性。

我通過更改為有DiGraph並確定sampler 1點頭的位置，對您實際嘗試做的事情做了一些假設。

這是一個獨立的代碼塊及其輸出。

import networkx as nx
import matplotlib.pyplot as plt


nodes = {'61697b94f74c92a808641ba3':'S3 connector',
'61697b95f74c92a808641ba4':'loader 1',
'61697b96f74c92a808641ba5':'loader 2',
'61697b96f74c92a808641ba6':'sampler 1',
'61697b97f74c92a808641ba7':'concator',
'61697b98f74c92a808641ba8':'sampler 2',
'61697b98f74c92a808641ba9':'splitter'}

edges = [('61697b94f74c92a808641ba3', '61697b95f74c92a808641ba4'),
('61697b94f74c92a808641ba3', '61697b96f74c92a808641ba5'),
('61697b95f74c92a808641ba4', '61697b96f74c92a808641ba6'),
('61697b96f74c92a808641ba6', '61697b97f74c92a808641ba7'),
('61697b96f74c92a808641ba5', '61697b97f74c92a808641ba7'),
('61697b97f74c92a808641ba7', '61697b98f74c92a808641ba8'),
('61697b98f74c92a808641ba8', '61697b98f74c92a808641ba9')]

layers = {'61697b94f74c92a808641ba3': 1,
'61697b95f74c92a808641ba4': 2,
'61697b96f74c92a808641ba5':2,
'61697b96f74c92a808641ba6':3,
'61697b97f74c92a808641ba7':4,
'61697b98f74c92a808641ba8':5,
'61697b98f74c92a808641ba9':6}

nx_graph = nx.DiGraph() # Made this a DiGraph (adds arrows to visual)
plt.figure(figsize=(8,8)) # Enlarged figure

for key, value in nodes.items():
    nx_graph.add_node(key, name=value, layer=layers[key])

for edge in edges:
    nx_graph.add_edge(*edge)

pos = nx.multipartite_layout(nx_graph, subset_key="layer")

nx.draw(nx_graph, pos=pos, labels=nodes, with_labels=True)

plt.show()