Qust Docs

DF / DataFrame 执行计划

Qust 的 DataFrame 是计算计划,不是数据容器。真实数据通常是 Polars DataFrame。

Python
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))