都営地下鉄の最短経路を図化

import pandas as pd

# 駅データ.jpの駅データを読み込み
df = pd.read_csv('station20200316free.csv')

# 都営地下鉄を抽出
df = df[df['line_cd'].isin([99301, 99302, 99303, 99304])][['station_name', 'lon', 'lat']]

# 押上を置き換え
df = df.replace('押上（スカイツリー前）', '押上')

# 重複データを削除
df = df.drop_duplicates()

# 出力
df.to_csv('pos.csv', encoding='shift-jis', index=None)

import openpyxl
import csv
import networkx as nx
import matplotlib.pyplot as plt

# 読み込み
lines = ['浅草線', '三田線', '新宿線', '大江戸線']
edge=[]
for line in lines:
    wb = openpyxl.load_workbook('distance.xlsx')
    sheet = wb[line]
    data = [[sheet['A' + str(i)].value, sheet['B' + str(i)].value] for i in range(1, sheet.max_row + 1)]
    
    for i in range(len(data)-1):
        edge.append([data[i][0], data[i + 1][0], abs(data[i][1]*1000 - data[i + 1][1]*1000)])

# 名前の違う接続駅を追加（重みは0）
edge.append(['東日本橋', '馬喰横山', 0])

# 座標データ読み込み
with open('pos.csv', 'r', encoding='shift-jis') as f:
    reader = csv.reader(f)
    next(reader)
    pos={row[0]: (float(row[1]), float(row[2])) for row in reader}

# グラフ作成
G = nx.Graph() # 無向グラフ
G.add_weighted_edges_from((e[0], e[1], e[2]) for e in edge)

# 描画
fig = plt.figure(figsize=(20, 20))
ax = fig.add_subplot(111)
nx.draw_networkx(G, ax=ax, pos=pos, font_family='MS Gothic', node_size=50)
plt.savefig('全駅表示.png')

# 検索経路
start = '両国'
goal = '御成門'

path = nx.dijkstra_path(G, start, goal)
edge_list = [(path[i], path[i+1]) for i in range(len(path)-1)]
weight = nx.dijkstra_path_length(G, start, goal)

# 描画
fig = plt.figure(figsize=(20, 20))
ax = fig.add_subplot(111)
nx.draw_networkx(G, ax=ax, pos=pos, font_family='MS Gothic', node_size=50, node_color='gray', with_labels=False)
nx.draw_networkx_nodes(G, pos, [start, goal ],node_color='red', node_size=100)
nx.draw_networkx_edges(G, pos, edge_list, edge_color='pink', width=8, alpha=0.8)
nx.draw_networkx_labels(G, pos, {start: start,  goal: goal}, font_family='MS Gothic', font_size=24)
nx.draw_networkx_labels(G, pos, {'蔵前': '蔵前', '東日本橋': '東日本橋\n馬喰横山', '大門': '大門', '三田': '三田',
                                 '神保町': '神保町', '春日': '春日', '新宿': '新宿', '森下': '森下', '都庁前': '都庁前'},
                                 font_family='MS Gothic', font_size=16, alpha=0.5)
plt.savefig('{}-{}.png'.format(start, goal))