Polars：在列名称中搜索和替换：是否可以使用 LazyFrames？

Question

已回答的问题的后续内容，是否可以在 LazyFrame 中搜索和替换列名称？我这样做是作为一种解决方法（基于 ritchie46 的链接答案，谢谢！）：

df = df.lazy().collect()
df.columns = list(map(lambda x: x.replace("Total", ""), df.columns)) 

Answer 1

还有

pl.LazyFrame.rename

。

import polars as pl

df = pl.LazyFrame({
    "Total_foo": [1],
    "bar_Total": [2],
    "other": [3],
})

df.rename(lambda name: name.replace("Total", "")).collect()

shape: (1, 3)
┌──────┬──────┬───────┐
│ _foo ┆ bar_ ┆ other │
│ ---  ┆ ---  ┆ ---   │
│ i64  ┆ i64  ┆ i64   │
╞══════╪══════╪═══════╡
│ 1    ┆ 2    ┆ 3     │
└──────┴──────┴───────┘

Answer 2

(pl.LazyFrame({
    "Total_foo": [1],
    "bar_Total": [2],
    "other": [3],
}).select(
     pl.all().name.map(lambda name: name.replace("Total", ""))
)).collect()

shape: (1, 3)
┌──────┬──────┬───────┐
│ _foo ┆ bar_ ┆ other │
│ ---  ┆ ---  ┆ ---   │
│ i64  ┆ i64  ┆ i64   │
╞══════╪══════╪═══════╡
│ 1    ┆ 2    ┆ 3     │
└──────┴──────┴───────┘