Каков самый быстрый способ выполнить «индексированный» поиск в Polars? ⇐ Python

[Anonymous]

Я работаю с большими кадрами данных Polars, которые полностью загружены в память. Каждая строка уникально индексируется столбцами entityId (Int64) и enterDate (дата).
Я знаю, что в poalars нет индексов, но мне все равно нужно выполнять специальный поиск данных по этим таблицам, и это достаточно часто, что на это уходит нетривиальный % времени выполнения моего приложения.
В настоящее время я нахожу эти строки с помощью .filter

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

def locate(df, entityId, entryDate)->pl.DataFrame:
return df.filter(pl.col('entityId')==entityId).filter(pl.col('entryDate') == entryDate)

Это довольно быстро, но поиск строки по-прежнему занимает от 50 до 100 мс.
Есть ли какие-либо оптимизации, которые мне не хватает?
Некоторые вещи, которые я пробовал:

• использование .lazy / .collect (без изменений)
• сортировка по идентификатору объекта (нет меняй)

Я на полярах 0.17.12

Подробнее здесь: https://stackoverflow.com/questions/762 ... -in-polars