Qust Docs

简单例子

Qust 的核心是把计算保留成表达式树,再交给执行计划运行。

Python
import polars as pl
import qust as qs
from qust import col

data = pl.DataFrame({
    "datetime": ["2026-01-01 09:30:00", "2026-01-01 09:31:00", "2026-01-01 09:32:00"],
    "code": ["AAA", "AAA", "AAA"],
    "close": [10.0, 10.2, 10.1],
}).with_columns(pl.col("datetime").str.to_datetime())

df = col.with_cols(
    col("close").pct().alias("ret"),
    col("close").ta.rsi(3).alias("rsi"),
).runtime()

print(df.calc_data(data))

注意事项

  • 真实数据建议先用 Polars 清理类型,再传给 calc_data
  • 字符串表示列名,数字通常表示 literal 或下标,具体看算子说明。
  • 需要参数联动时使用 pms(...),不要提前转成普通数字。