引用以前别名的Polar表达式

上下文 并行执行。因此,它们不能引用一个还不存在的列。

上下文是:

• df.with_columns
• df.select
• df.groupby(..).agg

这意味着您需要对引用其他表达式输出的表达式强制执行顺序。

在你的情况下,我会:

(df.with_column(
       (pl.col('x') + 1).alias('y')
    ).select([
       pl.col('y'),
      (pl.col('y') * 2).alias('z')
]))

一种解决方法是将每个新列都拉到自己的列中 with_column 打电话然后做期末考试 select 以保留您应该保留的列。你可能会想确保这是懒惰地完成。

import polars as pl

df = pl.DataFrame(dict(x=[0, 0, 1]))

(df
  .lazy()
  .with_column((pl.col("x") + 1).alias("y"))
  .with_column((pl.col("y") * 2).alias("z"))
  .select(["y", "z"])
  .collect()
)
# shape: (3, 2)
# ┌─────┬─────┐
# │ y   ┆ z   │
# │ --- ┆ --- │
# │ i64 ┆ i64 │
# ╞═════╪═════╡
# │ 1   ┆ 2   │
# ├╌╌╌╌╌┼╌╌╌╌╌┤
# │ 1   ┆ 2   │
# ├╌╌╌╌╌┼╌╌╌╌╌┤
# │ 2   ┆ 4   │
# └─────┴─────┘