【Python】＃2 後編 foliumで線をプロット -バス走行ルート地図のカスタマイズ-

import pandas as pd
import folium
from branca.element import Template, MacroElement


def draw_route(
    stops_file, stop_times_file, trips_file, routes_file,
    output_html="all_routes_map.html",
    show_markers=True
):
    # データ読み込み
    stops = pd.read_csv(stops_file)
    stop_times = pd.read_csv(stop_times_file)
    trips = pd.read_csv(trips_file)
    routes = pd.read_csv(routes_file)


    # tripsにルート名を結合
    trips = trips.merge(routes, on="route_id", how="left")


    # 地図の中心を全バス停の緯度経度の中央値
    lat_center = stops["stop_lat"].mean()
    lon_center = stops["stop_lon"].mean()
    m = folium.Map(location=[lat_center, lon_center], zoom_start=12)


    # 色リスト
    colors = [
    "red", "navy", "orange" ,"green", "blue", "purple",
    "deeppink", "gold", "magenta", "lime","brown"]


    unique_routes = trips["route_id"].unique()
    print(f"route_id数: {len(unique_routes)}件")


    legend_items = []


    for i, route_id in enumerate(unique_routes):
        route_trips = trips[trips["route_id"] == route_id]
        trip = route_trips.iloc[0]  # 最初のtripを使う
        trip_id = trip["trip_id"]
        route_name = trip.get('route_short_name', '').strip()
        color = colors[i % len(colors)]


        # バス停取得＆結合
        trip_stops = stop_times[stop_times["trip_id"] == trip_id].sort_values("stop_sequence")
        merged = trip_stops.merge(stops, on="stop_id")


        # 線の描画
        points = list(zip(merged["stop_lat"], merged["stop_lon"]))
        folium.PolyLine(
            points,
            color=color,
            weight= 4.5 ,
            opacity= 1 ,
            tooltip=f"{route_name}"
        ).add_to(m)


        # バス停にマーカーを追加する（円・ポップアップ設定）
        if show_markers:
            for _, row in merged.iterrows():
                popup_html = f"<p>{row['stop_name']}</>"
                folium.CircleMarker(
                    location=[row['stop_lat'], row['stop_lon']],
                    radius=5,
                    color='black',       # 枠線の色
                    weight=1.5,          # 枠線の太さ
                    fill=True,
                    fill_color=color,
                    fill_opacity=0.8,
                    popup=folium.Popup(popup_html, max_width=250)
                ).add_to(m)


        # 凡例用データ追加
        legend_items.append(f"<li style='color:{color}; padding:8px 0;'>{route_name}</li>")


    # 凡例HTML
    legend_html = f"""
    <div style="
        position: fixed;
        top: 20px;
        right: 20px;
        width: 280px;
        background: white;
        font-weight: bold;
        border:2px solid gray;
        border-radius:6px;
        z-index:9999;
        font-size:12px;
        box-shadow: 2px 2px 6px rgba(0,0,0,0.3);
    ">
    <div style="padding:6px;">
        <p>コース・路線名</p>
        <ul style="list-style:none;padding:0;margin:0;">
            {''.join(legend_items)}
        </ul>
    </div>
    </div>
    """


    # 凡例追加
    legend = MacroElement()
    legend._template = Template(f"""{{% macro html(this, kwargs) %}}{legend_html}{{% endmacro %}}""")
    m.get_root().add_child(legend)


    # 地図保存
    m.save(output_html)
    print(f"地図を保存しました: {output_html}")


# 使用例（ファイルパスは適宜変更）
draw_route(
    stops_file="stops.txt",
    stop_times_file="stop_times.txt",
    trips_file="trips.txt",
    routes_file="routes.txt"
)