Края MultiDiGraph от сети x отрисовывают с помощью connectionStyle - PullRequest
Новая
       42

Края MultiDiGraph от сети x отрисовывают с помощью connectionStyle

2 голосов
/ 05 февраля 2020

Можно ли как-то нарисовать разные ребра в одних и тех же узлах с разной кривизной, используя стиль соединения ?

Я написал следующий код, но все три ребра перекрылись: 

import networkx as nx
import matplotlib.pyplot as plt

G = nx.MultiDiGraph()
G.add_node('n1')
G.add_node('n2')
G.add_edge('n1', 'n2', 0)
G.add_edge('n1', 'n2', 1)
G.add_edge('n1', 'n2', 2)

pos = nx.spring_layout(G)
nx.draw(G, pos, with_labels=True, connectionstyle='arc3, rad = 0.3')

plt.show()

1 Ответ

2 голосов
/ 05 февраля 2020

Это может быть сделано путем построения каждого ребра с другим аргументом rad - как показано. Обратите внимание, что мой подход здесь использует f-строки, которые требуют Python 3.6 - ниже, что вам придется строить строку, используя другой метод.

Код: 

import networkx as nx
import matplotlib.pyplot as plt

G = nx.MultiDiGraph()
G.add_node('n1')
G.add_node('n2')
G.add_edge('n1', 'n2', rad=0.1)
G.add_edge('n1', 'n2', rad=0.2)
G.add_edge('n1', 'n2', rad=0.3)

plt.figure(figsize=(6,6))

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

for edge in G.edges(data=True):
    nx.draw_networkx_edges(G, pos, edgelist=[(edge[0],edge[1])], connectionstyle=f'arc3, rad = {edge[2]["rad"]}')

plt.show()

Вывод:

enter image description here

Мы могли бы даже создать новую функцию, чтобы сделать это для нас: 

import networkx as nx
import matplotlib.pyplot as plt

def new_add_edge(G, a, b):
    if (a, b) in G.edges:
        max_rad = max(x[2]['rad'] for x in G.edges(data=True) if sorted(x[:2]) == sorted([a,b]))
    else:
        max_rad = 0
    G.add_edge(a, b, rad=max_rad+0.1)

G = nx.MultiDiGraph()
G.add_node('n1')
G.add_node('n2')

for i in range(5):
    new_add_edge(G, 'n1', 'n2')

for i in range(5):
    new_add_edge(G, 'n2', 'n1')

plt.figure(figsize=(6,6))

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

for edge in G.edges(data=True):
    nx.draw_networkx_edges(G, pos, edgelist=[(edge[0],edge[1])], connectionstyle=f'arc3, rad = {edge[2]["rad"]}')

plt.show()

Выход:

enter image description here

...