Pythonにおけるdaijkstra法を用いてあるノード(x,y)の最短経路の長さを求める関数f(x,y)を作りたい

Question

前提・実現したいこと

daijkstra法を用いてあるノード(x,y)の最短経路の長さを求める関数f(x,y)を作りたい

発生している問題・エラーメッセージ

エラーメッセージ

Warning (from warnings module):
File "C:\Users\stuart\AppData\Local\Programs\Python\Python37\lib\site-packages\networkx\drawing\nx_pylab.py", line 579
if not cb.iterable(width):
MatplotlibDeprecationWarning:
The iterable function was deprecated in Matplotlib 3.1 and will be removed in 3.3. Use np.iterable instead.
Traceback (most recent call last):
File "F:\7.9卒論.py", line 78, in <module>
for source, mapping in all_pairs:
File "C:\Users\stuart\AppData\Local\Programs\Python\Python37\lib\site-packages\networkx\algorithms\shortest_paths\weighted.py", line 1049, in all_pairs_dijkstra_path_length
yield (n, length(G, n, cutoff=cutoff, weight=weight))
File "C:\Users\stuart\AppData\Local\Programs\Python\Python37\lib\site-packages\networkx\algorithms\shortest_paths\weighted.py", line 373, in single_source_dijkstra_path_length
weight=weight)
File "C:\Users\stuart\AppData\Local\Programs\Python\Python37\lib\site-packages\networkx\algorithms\shortest_paths\weighted.py", line 623, in multi_source_dijkstra_path_length
return _dijkstra_multisource(G, sources, weight, cutoff=cutoff)
File "C:\Users\stuart\AppData\Local\Programs\Python\Python37\lib\site-packages\networkx\algorithms\shortest_paths\weighted.py", line 831, in _dijkstra_multisource
vu_dist = dist[v] + cost
TypeError: unsupported operand type(s) for +: 'int' and 'str'

自作のプログラム

import networkx as nx
import matplotlib.pyplot as plt
import numpy as np

G=nx.Graph()

#ノードの座標の設定
pos={1:(0,4),
     2:(2,10),
     3:(4,4),
     4:(3,0),
     5:(10,11),
     6:(9,4),
     7:(11,0),
     8:(15,7),
     9:(13,2)}

nx.draw(G,pos)

#ユークリッド距離
x=pos
y=pos
def ans(x,y):
    vec = np.array(pos[x])-np.array(pos[y])
    dist1 = np.linalg.norm(vec)
    return f"{dist1:4.2f}"

#枝とその重みの設定
G.add_weighted_edges_from([(1,2,ans(1,2)),
                           (1,3,ans(1,3)),
                           (1,4,ans(1,4)),
                           (2,3,ans(2,3)),
                           (3,4,ans(3,4)),
                           (2,5,ans(2,5)),
                           (3,6,ans(3,6)),
                           (4,7,ans(4,7)),
                           (5,6,ans(5,6)),
                           (6,7,ans(6,7)),
                           (5,8,ans(5,8)),
                           (8,9,ans(8,9)),
                           (6,9,ans(6,9)),
                           (7,9,ans(7,9))])


#枝の長さをラベルとして設定
edge_labels={(1,2):ans(1,2),
             (1,3):ans(1,3),
             (1,4):ans(1,4),
             (2,3):ans(2,3),
             (3,4):ans(3,4),
             (2,5):ans(2,5),
             (3,6):ans(3,6),
             (4,7):ans(4,7),
             (5,6):ans(5,6),
             (6,7):ans(6,7),
             (5,8):ans(5,8),
             (8,9):ans(8,9),
             (6,9):ans(6,9),
             (7,9):ans(7,9)}

nx.draw_networkx_nodes(G,pos)
nx.draw_networkx_edges(G,pos,node_size=100,width=1)
nx.draw_networkx_edge_labels(G,pos,edge_labels)
nx.draw_networkx_labels(G,pos,font_size=12,font_color="w")

#x, y座標軸の定義
plt.xlim(-5.0, 20.0)
plt.xticks([])
plt.ylim(-5.0, 15.0)
plt.yticks([])
plt.show()

#daijkstra法
from networkx.algorithms.shortest_paths.weighted import all_pairs_dijkstra_path_length
all_pairs =all_pairs_dijkstra_path_length(G)
for source, mapping in all_pairs:
    for target in mapping.keys():
        if source != target:
            dist2 =mapping[target]
            print(f"({source}, {target}): {dist2:4.2f}")

試したこと

まず、関数f(x,y)を作る前に各ノード(x,y)の最短経路の長さと最短経路を表示させた

補足情報（FW/ツールのバージョンなど）

元々、ans(x,y)を作る前は表示されていたが「枝とその重みの設定」と「枝の長さをラベルとして設定させる」所でans(x,y)を呼び出したらエラーするようになった。

Comments

[bamboo-nova]

コードを載せるときは、なるべくmarkdown記法に習って```との間に入れるようにした方が答える方が見やすいと思うので、そうしてください。

Answer 1

def ans(x,y):
...
dist1 = np.linalg.norm(vec)
return f"{dist1:4.2f}"

余計に文字列化の処理を噛ませているのが問題では。
単に return dist1 で充分な気がします。

>>> n = 2
>>> 40 + n
42
>>> 40 + str(n)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: unsupported operand type(s) for +: 'int' and 'str'

コードの書き方について

teratailには、コードを見やすく表示する機能があります。
質問編集画面を開き、コードを選択した状態で<code>ボタンを押して下さい。

特にPythonの場合、インデントが崩れるとコードの意味が変わってしまいます。

Comments

[tiadia]

return dist1で最短経路の計算はできました！ありがとうございます。

ただ、この場合だとラベルとして表示させるdist1を小数点第二位までに収めたいのですが、それが出来なくなってしまいました。

[LouiS0616]

draw_networkx_edge_labelsにprecisionとかformatとかそういう引数があることを期待していたんですが、どうやら無さそうです。
https://networkx.github.io/documentation/networkx-1.10/reference/generated/networkx.drawing.nx_pylab.draw_networkx_edge_labels.html

---
networkxを使ったことが無いので遠回りな方法かもしれませんが、とりあえず次のように書けばエラー無く実行できました。結局一時的に文字列化させてます。

nx.draw_networkx_edge_labels(
G, pos,
edge_labels={k: '{:.2f}'.format(v) for k, v in edge_labels.items()}
)