foliumを使って新潟県県央地域の路線バスの1日の動きを可視化してみた #Python – Qiita

id,timestamp,latitude,longitude,route_id
10658301001,2025-03-21 06:58:00+09:00,37.66339075,138.8242534,0
...
49906,2025-03-21 06:23:48+09:00,37.61711121,138.8234711,1
...
20720101001,2025-03-21 07:20:00+09:00,37.63088791,138.973589,2

import pandas as pd
import folium
from folium.plugins import TimestampedGeoJson

# CSV読み込み
df = pd.read_csv('0321_echigo_tsubame_sanjo_gtfs_realtime_data.csv', parse_dates=['timestamp'])

# タイムゾーンを削除し、60秒間隔で時刻を丸める
df['timestamp'] = df['timestamp'].dt.tz_localize(None)
df['rounded_timestamp'] = df['timestamp'].dt.round('60s')

# 補間処理（丸め後）
interpolated_rows = []

for bus_id, group in df.groupby('id'):
    group = group.drop_duplicates('rounded_timestamp').set_index('rounded_timestamp').sort_index()
    group_resampled = group.reindex(pd.date_range(group.index.min(), group.index.max(), freq='60s'))

    group_resampled['id'] = bus_id
    group_resampled['route_id'] = group['route_id'].ffill()
    group_resampled[['latitude', 'longitude']] = group_resampled[['latitude', 'longitude']].interpolate()
    group_resampled = group_resampled.dropna(subset=['latitude', 'longitude', 'route_id'])

    # interpolated_rows.append(group_resampled.reset_index().rename(columns={'index': 'timestamp'}))
    interpolated_rows.append(group_resampled.reset_index().rename(columns={'index': 'rounded_timestamp'}))

df_interpolated = pd.concat(interpolated_rows).reset_index(drop=True)

# route_idごとの色設定
color_dict = {0.0: 'red', 1.0: 'blue', 2.0: 'yellow'}

# GeoJSON用データ展開
expanded_rows = []
for bus_id, group in df_interpolated.groupby('id'):
    group = group.sort_values('timestamp').reset_index(drop=True)
    for i in range(len(group)-1):
        current_row = group.iloc[i].copy()
        current_row['next_timestamp'] = group.iloc[i+1]['timestamp']
        expanded_rows.append(current_row)

df_expanded = pd.DataFrame(expanded_rows)

# GeoJSON生成
def df_to_geojson(df):
    features = []
    for _, row in df.iterrows():
        color = color_dict[row['route_id']]
        feature = {
            'type': 'Feature',
            'geometry': {
                'type': 'Point',
                'coordinates': [row['longitude'], row['latitude']],
            },
            'properties': {
                'times': [
                    row['timestamp'].strftime('%Y-%m-%dT%H:%M:%S'),
                    row['next_timestamp'].strftime('%Y-%m-%dT%H:%M:%S')
                ],
                'icon': 'circle',
                'iconstyle': {
                    'fillColor': color,
                    'color': color,
                    'fillOpacity': 0.8,
                    'stroke': 'true',
                    'radius': 6
                },
                'popup': (
                    f"バスID: {row['id']}
"
                    f"路線ID: {int(row['route_id'])}
"
                    f"時刻: {row['timestamp'].strftime('%Y-%m-%d %H:%M:%S')}"
                )
            }
        }
        features.append(feature)
    return {'type': 'FeatureCollection', 'features': features}

geojson_data = df_to_geojson(df_expanded)

# 地図作成
m = folium.Map(location=[df['latitude'].mean(), df['longitude'].mean()], zoom_start=13)

# Foliumアニメーション設定
TimestampedGeoJson(
    geojson_data,
    period='PT1M',  # 60秒間隔
    add_last_point=False,
    auto_play=True,
    loop=False,
    max_speed=120,  # 2分を1秒に高速再生
    loop_button=True,
    date_options='YYYY/MM/DD HH:mm:ss',
    time_slider_drag_update=True,
    duration='PT1M'
).add_to(m)

# 凡例表示
legend_html = '''
"position: fixed; bottom: 50px; left: 30px; width: 120px; z-index:9999; background-color:white; padding:10px; border:2px solid grey; border-radius:6px;">
凡例
"background:red;width:10px;height:10px;border-radius:50%;display:inline-block;"> Route 0
"background:blue;width:10px;height:10px;border-radius:50%;display:inline-block;"> Route 1
"background:yellow;width:10px;height:10px;border-radius:50%;display:inline-block;"> Route 2

'''
m.get_root().html.add_child(folium.Element(legend_html))

# HTMLとして保存
m.save('bus_animation.html')

print("アニメーションHTML生成完了: bus_animation.html")