FastF1でドライバーのテレメトリーを比較する方法【Python・コピペOK】

import fastf1
import fastf1.plotting
import matplotlib.pyplot as plt
import os

# キャッシュ用フォルダ作成
os.makedirs('cache', exist_ok=True)

# キャッシュ機能を有効化
fastf1.Cache.enable_cache('cache')

# Matplotlib設定（時間形式対応・FastF1配色・チームカラーを使用）
fastf1.plotting.setup_mpl(mpl_timedelta_support=True, color_scheme='fastf1')

# セッション読み込み（年：2026, 開催国：オーストラリア, セッション：予選）
session = fastf1.get_session(2026, "Australia", "Q")
session.load(telemetry=True)

# =========================
# ① ドライバーを指定
# =========================
driver = "RUS"  # ラッセル

# =========================
# ② 最速ラップの取得
# =========================
lap = session.laps.pick_drivers(driver).pick_fastest()

# テレメトリーデータ取得（距離も追加）
tel = lap.get_car_data().add_distance()

# =========================
# ③ チームカラーを取得
# =========================
team_name = lap['Team']  # 例: Mercedes
color = fastf1.plotting.get_team_color(team_name, session=session)

# =========================
# ④ コーナー情報を取得
# =========================
circuit = session.get_circuit_info()
corners = circuit.corners

# =========================
# ⑤ グラフ作成
# =========================
fig, ax = plt.subplots(4, 1, figsize=(11, 10), sharex=True)

# --- Speed ---
ax[0].plot(tel['Distance'], tel['Speed'], color=color)
ax[0].set_ylabel("Speed (km/h)")
ax[0].grid(axis="y", linestyle="--", alpha=0.5)

# --- Throttle ---
ax[1].plot(tel['Distance'], tel['Throttle'], color=color)
ax[1].set_ylabel("Throttle (%)")

# --- Brake ---
ax[2].plot(tel['Distance'], tel['Brake'], color=color)
ax[2].set_ylabel("Brake")

# --- Gear ---
ax[3].plot(tel['Distance'], tel['nGear'], color=color)
ax[3].set_ylabel("Gear")
ax[3].set_xlabel("Distance (m)")
ax[3].set_ylim(0.5, 8.5)
ax[3].set_yticks(range(1, 9))
ax[3].grid(axis="y", linestyle="--", alpha=0.5)

# =========================
# ⑥ コーナー位置を表示
# =========================
for _, corner in corners.iterrows():
    dist = corner['Distance']
    number = corner['Number']

    for a in ax:
        a.axvline(dist, color='gray', linestyle=':', alpha=0.5)

    ax[0].text(
        dist,
        ax[0].get_ylim()[1],
        f"T{number}",
        rotation=90,
        verticalalignment='bottom',
        fontsize=8,
        color='gray'
    )

# =========================
# ⑦ タイトル
# =========================
plt.suptitle(f"{session.event['EventName']} - {driver} Fastest Lap Telemetry")

plt.tight_layout()
plt.show()

# =========================
# ① ドライバーを指定
# =========================
driver1 = "RUS"  # ラッセル
driver2 = "HAM"  # ハミルトン

# =========================
# ② 最速ラップの取得
# =========================
lap1 = session.laps.pick_drivers(driver1).pick_fastest()
lap2 = session.laps.pick_drivers(driver2).pick_fastest()

# テレメトリーデータ取得（距離も追加）
tel1 = lap1.get_car_data().add_distance()
tel2 = lap2.get_car_data().add_distance()

# =========================
# ③ チームカラーを取得
# =========================
color1 = fastf1.plotting.get_team_color(lap1['Team'], session=session)
color2 = fastf1.plotting.get_team_color(lap2['Team'], session=session)

# =========================
# ④ コーナー情報を取得
# =========================
circuit = session.get_circuit_info()
corners = circuit.corners

# =========================
# ⑤ グラフ作成
# =========================
fig, ax = plt.subplots(4, 1, figsize=(11, 10), sharex=True)

# --- Speed ---
ax[0].plot(tel1['Distance'], tel1['Speed'], color=color1, label=driver1)
ax[0].plot(tel2['Distance'], tel2['Speed'], color=color2, label=driver2)
ax[0].set_ylabel("Speed (km/h)")
ax[0].grid(axis="y", linestyle="--", alpha=0.5)
ax[0].legend()

# --- Throttle ---
ax[1].plot(tel1['Distance'], tel1['Throttle'], color=color1, label=driver1)
ax[1].plot(tel2['Distance'], tel2['Throttle'], color=color2, label=driver2)
ax[1].set_ylabel("Throttle (%)")
ax[1].legend()

# --- Brake ---
ax[2].plot(tel1['Distance'], tel1['Brake'], color=color1, label=driver1)
ax[2].plot(tel2['Distance'], tel2['Brake'], color=color2, label=driver2)
ax[2].set_ylabel("Brake")
ax[2].legend()

# --- Gear ---
ax[3].plot(tel1['Distance'], tel1['nGear'], color=color1, label=driver1)
ax[3].plot(tel2['Distance'], tel2['nGear'], color=color2, label=driver2)
ax[3].set_ylabel("Gear")
ax[3].set_xlabel("Distance (m)")
ax[3].set_ylim(0.5, 8.5)
ax[3].set_yticks(range(1, 9))
ax[3].grid(axis="y", linestyle="--", alpha=0.5)
ax[3].legend()

# =========================
# ⑥ コーナー位置を表示
# =========================
for _, corner in corners.iterrows():
    dist = corner['Distance']
    number = corner['Number']

    for a in ax:
        a.axvline(dist, color='gray', linestyle=':', alpha=0.5)

    ax[0].text(
        dist,
        ax[0].get_ylim()[1],
        f"T{number}",
        rotation=90,
        verticalalignment='bottom',
        fontsize=8,
        color='gray'
    )

# =========================
# ⑦ タイトル
# =========================
plt.suptitle(f"{session.event['EventName']} - {driver1} vs {driver2} Fastest Lap Telemetry")

plt.tight_layout()
plt.show()