Как я могу эффективно заполнить_null только определенные столбцы DataFrame? ⇐ Python

[Anonymous]

Например, предположим, что я хочу заполнить_null(strategy="zero") только числовые столбцы моего DataFrame. Моя текущая стратегия состоит в следующем:

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

import polars as pl
import polars.selectors as cs

df = pl.DataFrame(
[
pl.Series("id", ["alpha", None, "gamma"]),
pl.Series("xs", [None, 100, 2]),
]
)

final_df = df.select(cs.exclude(cs.numeric()))
final_df = final_df.with_columns(
df.select(cs.numeric()).fill_null(strategy="zero")
)

print(final_df)

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

shape: (3, 2)
┌───────┬─────┐
│ id    ┆ xs  │
│ ---   ┆ --- │
│ str   ┆ i64 │
╞═══════╪═════╡
│ alpha ┆ 0   │
│ null  ┆ 100 │
│ gamma ┆ 2   │
└───────┴─────┘

Есть ли альтернативные, более идиоматические или более эффективные методы достижения того, чего я хочу?


Подробнее здесь: https://stackoverflow.com/questions/787 ... -dataframe