Лучший способ отфильтровать фрейм данных Polars для поиска элементов в нескольких столбцах ⇐ Python

[Anonymous]

Я хотел бы знать элегантный способ фильтрации фрейма данных на основе условия, что элементы в списке находятся в нескольких столбцах фрейма данных. Например, я хочу фильтровать df на основе всех элементов с to_keep, находящихся в столбцах c1 и c2.
import polars as pl

df = pl.DataFrame({"id": range(11)})

df = df.with_columns(
pl.format("row_{}", "id"),
pl.linear_space(0, 20, 11).alias("c1"),
pl.linear_space(0, 10, 11).alias("c2")
)

to_keep = [5, 7, 16, 18]

cond1 = df['c1'].map_elements(lambda ser: True if ser in to_keep else False)
cond2 = df['c2'].map_elements(lambda ser: True if ser in to_keep else False)

print(df.filter(cond1 | cond2))

Возвращает следующее:
shape: (4, 3)
┌───────┬──────┬─────┐
│ id ┆ c1 ┆ c2 │
│ --- ┆ --- ┆ --- │
│ str ┆ f64 ┆ f64 │
╞═══════╪══════╪═════╡
│ row_5 ┆ 10.0 ┆ 5.0 │
│ row_7 ┆ 14.0 ┆ 7.0 │
│ row_8 ┆ 16.0 ┆ 8.0 │
│ row_9 ┆ 18.0 ┆ 9.0 │
└───────┴──────┴─────┘


Подробнее здесь: https://stackoverflow.com/questions/771 ... ple-column