简单例子
Qust 的核心是把计算保留成表达式树,再交给执行计划运行。
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(...),不要提前转成普通数字。