Сравните две таблицы в Polars и выберите значение на основе этого сравнения. ⇐ Python

[Anonymous]

У меня есть такая таблица в полярах:

Код: Выделить всё

df_a = pl.from_repr("""
┌─────────────────────┬───────┐
│ arrival_time        ┆ Train │
│ ---                 ┆ ---   │
│ datetime[ns]        ┆ i64   │
╞═════════════════════╪═══════╡
│ 2024-10-04 08:40:10 ┆ 112   │
│ 2024-10-04 19:31:26 ┆ 134   │
└─────────────────────┴───────┘
""")

У меня есть еще одна таблица, которая определяет период дня на основе часов:

Код: Выделить всё

df_b = pl.from_repr("""
┌─────────────────────┬───────────┐
│ Time                ┆ Period    │
│ ---                 ┆ ---       │
│ datetime[ns]        ┆ str       │
╞═════════════════════╪═══════════╡
│ 2024-10-04 08:00:00 ┆ Early     │
│ 2024-10-04 16:00:00 ┆ Afternoon │
└─────────────────────┴───────────┘
""")

Я пытаюсь достичь способа Polars объединить обе таблицы и получить в первой таблице период дня в виде нового столбца:

Код: Выделить всё

┌─────────────────────┬───────┬───────────┐
│ arrival_time        ┆ Train ┆ Period    │
│ ---                 ┆ ---   ┆ ---       │
│ datetime[ns]        ┆ i64   ┆ str       │
╞═════════════════════╪═══════╪═══════════╡
│ 2024-10-04 08:40:10 ┆ 112   ┆ Early     │
│ 2024-10-04 19:31:26 ┆ 134   ┆ Afternoon │
└─────────────────────┴───────┴───────────┘

Сейчас я полностью работаю со словарями, архивирую два столбца моей таблицы сравнения и вычисляю ключ минимального расстояния между двумя столбцами времени:

Код: Выделить всё

min(dict(zip( df.Period,df.Time)).items(), key=lambda x: abs(pl.col('arrival_time') - x[1]))[0])

Но я определенно уверен, что в Polars есть лучший способ обработки.

Подробнее здесь: https://stackoverflow.com/questions/726 ... comparison