Легенда диаграммы Альтаира для подмножества данных ⇐ Python

[Anonymous]

В качестве упражнения для изучения более продвинутого альтаира я пытаюсь создать упрощенную версию этой диаграммы: https://climatereanalyzer.org/clim/t2_d ... m_id=world.
Для упрощения я использую серый цвет для всех лет до 2023 года, а затем красный и черный для 2023 и 2024 годов соответственно. Мне бы хотелось иметь легенду, которая бы относилась либо только к 2023 и 2024 годам, либо к «1940-2022», «2023», «2024».
Сейчас я сосредоточен на получить компактную легенду, отражающую любую подгруппу лет, но я бы прислушался к любому совету о том, как улучшить код/подход.
import pandas as pd
import altair as alt

# Function to fetch and prepare the data
def fetch_and_prep_data():
url = "https://climatereanalyzer.org/clim/t2_d ... 2_day.json"
data = requests.get(url).json()

years = []
all_temperatures = []

for year_data in data:
year = year_data['name']
temperatures = year_data['data']
temperatures = [temp if temp is not None else float('nan') for temp in temperatures]
days = list(range(1, len(temperatures) + 1))

df = pd.DataFrame({
'Year': [year] * len(temperatures),
'Day': days,
'Temperature': temperatures
})

years.append(year)
all_temperatures.append(df)

df_at = pd.concat(all_temperatures)

# Drop all rows where Year is more than 4 digits
df_at = df_at[df_at['Year'].str.len()

Подробнее здесь: https://stackoverflow.com/questions/786 ... et-of-data