表达式核心 / expr.py

expr.get_by_index

重排行算子，按参数表达式给出的 `UInt32` 行号，从当前输入里取行。

接口示例returns: Exprexpr

输入 / 输出

输入

price, open_sig, exit_sig

输入项	类型	示例
`price`	`Float64`	10.0
`open_sig`	`Boolean`	False
`exit_sig`	`Boolean`	False

输出

项目	说明
返回类型	`Expr`
输出对象	表达式/执行计划/配置对象
输出语义	输出列由算子、alias 或底层实现决定；需要稳定列名时显式使用 alias。
执行方式	先构造对象，再放入 DataSet、Monitor、UDF 或真实执行上下文。
核心调用	`col("price").get_by_index(col("open_sig", "exit_sig").stra.trade_row(), keep=True)`

打印输入 / 打印输出

下面内容来自本页示例代码真实执行后的 stdout，不是手写占位。

打印输入

print(data)

shape: (20, 3)
┌───────┬──────────┬──────────┐
│ price ┆ open_sig ┆ exit_sig │
│ ---   ┆ ---      ┆ ---      │
│ f64   ┆ bool     ┆ bool     │
╞═══════╪══════════╪══════════╡
│ 10.0  ┆ false    ┆ false    │
│ 10.2  ┆ false    ┆ false    │
│ 10.1  ┆ true     ┆ false    │
│ 10.5  ┆ false    ┆ false    │
│ 10.7  ┆ false    ┆ true     │
│ …     ┆ …        ┆ …        │
│ 12.4  ┆ false    ┆ false    │
│ 12.2  ┆ false    ┆ false    │
│ 12.7  ┆ false    ┆ true     │
│ 12.9  ┆ true     ┆ false    │
│ 13.1  ┆ false    ┆ false    │
└───────┴──────────┴──────────┘

打印输出

print(out)

shape: (4, 2)
┌────────────────┬─────────────────────────────────┐
│ 项目           ┆ 内容                            │
│ ---            ┆ ---                             │
│ str            ┆ str                             │
╞════════════════╪═════════════════════════════════╡
│ 调用           ┆ col("price").get_by_index(col(… │
│ 返回类型       ┆ Expr                            │
│ 状态           ┆ 表达式构造完成                  │
│ 怎么得到业务表 ┆ 放进                            │
│                ┆ col(...).runtime().calc_dat…    │
└────────────────┴─────────────────────────────────┘

调用

Python

col("price").get_by_index(col("open_sig", "exit_sig").stra.trade_row(), keep=True)

参数	类型	默认值	说明
`index`	`ExprInput`	`必填`	位置参数
`keep`	`bool`	`False`	位置参数

完整代码

这个算子页使用接口示例：不伪造计算结果。需要真实上下文、多输入源、Monitor session、UDF 回调、策略状态，或当前底层实现之后再执行。

展开可复制完整代码

Python

import datetime as dt
import polars as pl
import qust as qs
from qust import col, pms

data = pl.DataFrame(
    {
        "price": [10.0, 10.2, 10.1, 10.5, 10.7, 10.4, 10.9, 11.1, 10.8, 11.3, 11.6, 11.4, 11.9, 12.1, 11.8, 12.4, 12.2, 12.7, 12.9, 13.1],
        "open_sig": [False, False, True, False, False, True, False, False, False, True, False, False, False, False, True, False, False, False, True, False],
        "exit_sig": [False, False, False, False, True, False, False, False, True, False, False, False, False, True, False, False, False, True, False, False],
    }
)

pool = qs.DataPool("doc_pool")
monitor = qs.Monitor()
right = qs.DataSource("right")
schema = [("x", qs.dt.Float64)]
plot_expr = col("x")
df = col("x").runtime()

print("算子:")
print('expr.get_by_index')
print("场景:")
print('表达式核心：在执行计划里组合计算、窗口、过滤或元数据操作。')
print("模式:")
print('接口示例：只构造表达式或对象，不伪造计算结果。')
print("输入列:")
print('price, open_sig, exit_sig')
print("调用:")
print('col("price").get_by_index(col("open_sig", "exit_sig").stra.trade_row(), keep=True)')
print("输入数据模板:")
print(data)
try:
    result = col("price").get_by_index(col("open_sig", "exit_sig").stra.trade_row(), keep=True)
except BaseException as err:
    out = pl.DataFrame({
        "项目": ["调用", "状态", "错误类型", "错误信息"],
        "内容": ['col("price").get_by_index(col("open_sig", "exit_sig").stra.trade_row(), keep=True)', "未执行成数据表", type(err).__name__, str(err)[:200]],
    })
    print("输出:")
    print(out)
else:
    out = pl.DataFrame({
        "项目": ["调用", "返回类型", "状态", "怎么得到业务表"],
        "内容": ['col("price").get_by_index(col("open_sig", "exit_sig").stra.trade_row(), keep=True)', type(result).__name__, "表达式构造完成", "放进 col(...).runtime().calc_data(data)、col.with_cols(...).runtime().calc_data(data)，或对应 Monitor/DataSet 运行上下文。"],
    })
    print("输出:")
    print(out)

改成业务代码

改哪里	怎么改
列名	把示例 DataFrame 里的列名换成你的真实列名，列顺序保持和用法一致。
参数	只改函数括号里的参数；不要随意改变 `rolling/over/batch/select` 的链式层级。
输出名	需要稳定输出列名时，在表达式尾部加 `.alias("name")`。
调试	先打印输入数据和调用字符串，再执行 `calc_data`；报 schema 错时先检查列数和 dtype。

注意事项

- `index` 没有以前缀 `UInt32` 列开头：会报输入类型错误
- `index` 含 `null`：会直接报错
- 普通上下文下，`index` 必须是整条输入累计行号
- 外层 `over(...)` 上下文下，`index` 必须是同一 group 内的组内行号
- `index` 子表达式自己带 `over(...)` 会报错；请把 `over(...)` 写在外层

来源

项目	位置
源码文件	`expr.py`
类/对象	`Expr`