DF / DataFrame 执行计划
Qust 的 DataFrame 是计算计划,不是数据容器。真实数据通常是 Polars DataFrame。
import polars as pl
import qust as qs
from qust import col
data = pl.DataFrame({"x": [1.0, 2.0, 3.0], "y": [3.0, 2.0, 1.0]})
df = col((col("x") + col("y")).alias("sum_xy")).runtime()
print(df.calc_data(data))
df2 = col.with_cols((col("x") + col("y")).alias("sum_xy")).with_cols(
col("sum_xy").mean().expanding().alias("cum_mean")
).runtime()
print(df2.calc_data(data))