基於其他列創建新的 polars dataframe 列

Question

我很難理解 polars 數據幀上應該是一個非常常見的用例。 我只是想根據其他列在現有 dataframe 上創建一個新列。 這是我嘗試但不起作用的代碼：

import polars as pl
df.with_columns([
                (pl.col('old_col').apply(lambda x: func(x)).alias("new_col"),
            ])

Answer 1

Polars 使用純函數，這意味着它們沒有任何副作用，也不會可變地改變self 。

此屬性將確保您可以安全地將DataFrame傳遞給另一個 function，而不必擔心 function 會更改您下面的 DataFrame 的DataFrame 。

如果要更改分配的變量，只需重新分配新的 state 即可：

df = df.with_column(...)