Способы определения сезона по дате в языке программирования Python

Расширенные примеры и нюансы

1. Универсальная функция с поддержкой разных форматов

from datetime import datetime

def get_season(date_input, fmt=None):
    """
    Возвращает название сезона для переданной даты.
    date_input: строка, datetime или date
    fmt: формат строки (если не указан, используется %Y-%m-%d)
    """
    dt = None
    if isinstance(date_input, str):
        if fmt is None:
            fmt = "%Y-%m-%d"
        dt = datetime.strptime(date_input, fmt)
    elif isinstance(date_input, datetime):
        dt = date_input
    elif hasattr(date_input, 'month'):  # object with .month attribute
        dt = date_input
    else:
        raise ValueError("Неверный тип даты")
    
    month = dt.month
    day = dt.day if hasattr(dt, 'day') else 1
    
    # метеорологические сезоны (начиная с 1 числа)
    if (month == 12) or (month <= 2):
        return 'Зима'
    elif (month == 3) or (month == 4) or (month == 5):
        return 'Весна'
    elif (month == 6) or (month == 7) or (month == 8):
        return 'Лето'
    else:
        return 'Осень'

# Примеры вызова
print(get_season("2023-12-31"))  # Зима
print(get_season(datetime(2023, 6, 20)))  # Лето
print(get_season("31.12.2023", "%d.%m.%Y"))  # Зима

Зима
Лето
Зима

2. Обработка списка дат с выводом в DataFrame

import pandas as pd
from datetime import datetime

dates = ["2023-02-28", "2023-03-01", "2023-06-01", "2023-09-01", "2023-11-30"]
df = pd.DataFrame({'date_str': dates})
df['date'] = pd.to_datetime(df['date_str'])
df['season'] = df['date'].apply(get_season)
print(df)

    date_str       date season
0  2023-02-28 2023-02-28   Зима
1  2023-03-01 2023-03-01  Весна
2  2023-06-01 2023-06-01   Лето
3  2023-09-01 2023-09-01   Осень
4  2023-11-30 2023-11-30   Осень

3. Учёт високосного года (29 февраля)

# 29 февраля существует, и это всегда зима (метеорологическая)
print(get_season("2024-02-29"))  # Зима
# Проверка границ марта в високосный год
print(get_season("2024-03-01"))  # Весна

Зима
Весна

4. Астрономический сезон с точностью до часа

import ephem
from datetime import datetime, timezone

# Используем UTC, чтобы избежать часовых поясов
def astro_season_utc(dt):
    year = dt.year
    v = ephem.next_vernal_equinox(str(year))
    s = ephem.next_summer_solstice(v)
    a = ephem.next_autumnal_equinox(s)
    w = ephem.next_winter_solstice(a)
    
    # Преобразуем ephem-даты в datetime (UTC)
    v_dt = ephem.localtime(v).replace(tzinfo=timezone.utc)
    s_dt = ephem.localtime(s).replace(tzinfo=timezone.utc)
    a_dt = ephem.localtime(a).replace(tzinfo=timezone.utc)
    w_dt = ephem.localtime(w).replace(tzinfo=timezone.utc)
    
    if dt < v_dt:
        return 'Зима'
    elif dt < s_dt:
        return 'Весна'
    elif dt < a_dt:
        return 'Лето'
    elif dt < w_dt:
        return 'Осень'
    else:
        return 'Зима'

test = datetime(2023, 6, 21, 14, 57, tzinfo=timezone.utc)  # около летнего солнцестояния
print(astro_season_utc(test))

Лето

5. Определение сезона с помощью сдвига месяца (альтернативный подход)

def season_shift(dt):
    # Сдвиг на 2 месяца (март = 5, апрель = 6 и т.д.)
    # Декабрь (12) + 1 = 13 -> 13 % 12 = 1 -> зима
    # Не очень наглядно, но компактно
    return ['Зима', 'Весна', 'Лето', 'Осень'][((dt.month + 1) % 12) // 3]

dates_test = [datetime(2023, 1, 1), datetime(2023, 4, 1), datetime(2023, 7, 1), datetime(2023, 10, 1)]
for d in dates_test:
    print(d.date(), season_shift(d))

2023-01-01 Зима
2023-04-01 Весна
2023-07-01 Лето
2023-10-01 Осень

Важно:

6. Работа с часовыми поясами при астрономическом подходе

import ephem
from datetime import datetime, timezone, timedelta

# Если даты в локальном времени, нужно привести к UTC
def astro_season_local(dt_local, utc_offset_hours=3):
    dt_utc = dt_local - timedelta(hours=utc_offset_hours)
    return astro_season_utc(dt_utc)

# Пример: Москва (UTC+3) 20 марта 2023 10:00
local_dt = datetime(2023, 3, 20, 10, 0)
print(astro_season_local(local_dt, 3))  # может быть Весна или Зима в зависимости от точного времени

Результат может отличаться от метеорологического - астрономическая весна 2023 года наступила 20 марта в 21:24 UTC, поэтому в Москве в 10:00 ещё была зима.