networkxグラフでエッジの重みの位置を取得するにはどうすればよいですか?

Aug 25 2020

現在、networkxライブラリにはすべてのノードの位置を取得するための関数がありますspring_layout。ドキュメントから引用すると、次のようになります。

dict:ノードによってキー設定された位置の辞書

そして、次のように使用できます。

G=nx.path_graph(4)
pos = nx.spring_layout(G)

重み付きグラフのエッジウェイトの位置アクセスするのと同じようなものが欲しいのですが。エッジウェイトの数値が配置される位置、できればエッジの中央とエッジのすぐ上に戻る必要があります。(上記では、グラフの「外側」を意味するため、水平に配置された正方形のグラフの最下部のエッジの場合、エッジのすぐ下になります)。

だから問題は、spring_layoutこれを達成するために似たようなものが組み込まれているのかということです。そうでない場合は、自分でそれをどのように行うのですか?

回答

1 MohammedKashif Aug 25 2020 at 01:33

nx.draw_edge_labelsエッジをキーおよび(x, y, label)値として持つ辞書を返すwhichを使用できます

import matplotlib.pyplot as plt
import networkx as nx

# Create a graph
G = nx.path_graph(10)

# Add 2 egdes with labels
G.add_edge(0, 8, name='n1')
G.add_edge(2, 7, name='n2')

# Get the layout
pos = nx.spring_layout(G)

# Draw the graph
nx.draw(G, pos=pos)

# Draw the edge labels
edge_labels = nx.draw_networkx_edge_labels(G, pos)

今、あなたは変数を見ることができます edge_labels

print(edge_labels)
# {(0, 1): Text(0.436919941201627, -0.2110471432994752, '{}'),
#  (0, 8): Text(0.56941037628304, 0.08059107891826373, "{'name': 'n1'}"),
#  (1, 2): Text(0.12712625526483384, -0.2901338796021985, '{}'),
#  (2, 3): Text(-0.28017240645783603, -0.2947104829441387, '{}'),
#  (2, 7): Text(0.007024254096114596, -0.029867791669433513, "{'name': 'n2'}"),
#  (3, 4): Text(-0.6680363649371021, -0.26708812849092933, '{}'),
#  (4, 5): Text(-0.8016944207643129, -0.0029986274715349814, '{}'),
#  (5, 6): Text(-0.5673817462107436, 0.23808073918504968, '{}'),
#  (6, 7): Text(-0.1465270298295821, 0.23883392944036055, '{}'),
#  (7, 8): Text(0.33035539545007536, 0.2070939421162053, '{}'),
#  (8, 9): Text(0.7914739158501038, 0.2699223242747882, '{}')}

今、例えばエッジの位置を取得するには(2,7)、あなたはただする必要があります

print(edge_labels[(2,7)].get_position())
# Output: (0.007024254096114596, -0.029867791669433513)

ドキュメントの詳細については、こちらをご覧ください。

x,yすべてのエッジの座標を抽出する場合は、次のことを試すことができます。

edge_label_pos = { k: v.get_position()
                  for k, v in edge_labels.items()}
#{(0, 1): (0.436919941201627, -0.2110471432994752),
# (0, 8): (0.56941037628304, 0.08059107891826373),
# (1, 2): (0.12712625526483384, -0.2901338796021985),
# (2, 3): (-0.28017240645783603, -0.2947104829441387),
# (2, 7): (0.007024254096114596, -0.029867791669433513),
# (3, 4): (-0.6680363649371021, -0.26708812849092933),
# (4, 5): (-0.8016944207643129, -0.0029986274715349814),
# (5, 6): (-0.5673817462107436, 0.23808073918504968),
# (6, 7): (-0.1465270298295821, 0.23883392944036055),
# (7, 8): (0.33035539545007536, 0.2070939421162053),
# (8, 9): (0.7914739158501038, 0.2699223242747882)}