Заполнение нулевых значений столбца другим столбцом

Я только что нашел возможное решение:

df.with_column(
    pl.when(pl.col("c").is_null())
    .then(pl.col("b"))
    .otherwise(pl.col("a")).alias("a")
)

Для этого есть функция: fill_null.

Допустим, у нас есть эти данные:

import polars as pl

df = pl.DataFrame({'a': [1, None, 3, 4],
                   'b': [10, 20, 30, 40]
                   }).lazy()
print(df.collect())

shape: (4, 2)
┌──────┬─────┐
│ a    ┆ b   │
│ ---  ┆ --- │
│ i64  ┆ i64 │
╞══════╪═════╡
│ 1    ┆ 10  │
├╌╌╌╌╌╌┼╌╌╌╌╌┤
│ null ┆ 20  │
├╌╌╌╌╌╌┼╌╌╌╌╌┤
│ 3    ┆ 30  │
├╌╌╌╌╌╌┼╌╌╌╌╌┤
│ 4    ┆ 40  │
└──────┴─────┘

Мы можем заполнить нулевые значения в столбце a значениями в столбце b:

df.with_column(pl.col('a').fill_null(pl.col('b'))).collect()

shape: (4, 2)
┌─────┬─────┐
│ a   ┆ b   │
│ --- ┆ --- │
│ i64 ┆ i64 │
╞═════╪═════╡
│ 1   ┆ 10  │
├╌╌╌╌╌┼╌╌╌╌╌┤
│ 20  ┆ 20  │
├╌╌╌╌╌┼╌╌╌╌╌┤
│ 3   ┆ 30  │
├╌╌╌╌╌┼╌╌╌╌╌┤
│ 4   ┆ 40  │
└─────┴─────┘

Производительность этого будет довольно хорошей.