批内股票命名空间 / batch_stock.py
batch_stock.bt_intraday
股票日内换仓回测(先卖后买,含 T+1 与涨跌停/异常约束)。
接口示例returns: Exprbatch_stock
输入 / 输出
输入
按下表列名和类型准备输入。
| 输入项 | 类型 | 示例 |
|---|---|---|
datetime | Datetime | 2026-01-01 09:30:00 |
code | String | AAA |
w | Float64 | 1.0 |
price | Float64 | 10.0 |
flag | Boolean | False |
limit_up | Float64 | 11.0 |
limit_down | Float64 | 9.0 |
输出
| 项目 | 说明 |
|---|---|
| 返回类型 | Expr |
| 输出对象 | 表达式/执行计划/配置对象 |
| 输出语义 | 输出列由算子、alias 或底层实现决定;需要稳定列名时显式使用 alias。 |
| 执行方式 | 先构造对象,再放入 DataSet、Monitor、UDF 或真实执行上下文。 |
| 核心调用 | col("datetime", "code", "w", "price", "flag", "limit_up", "limit_down").batch_stock.bt_intraday(100.0, 0.0003) |
打印输入 / 打印输出
下面内容来自本页示例代码真实执行后的 stdout,不是手写占位。
打印输入
shape: (20, 7) ┌─────────────────────┬──────┬───────┬───────┬───────┬──────────┬────────────┐ │ datetime ┆ code ┆ w ┆ price ┆ flag ┆ limit_up ┆ limit_down │ │ --- ┆ --- ┆ --- ┆ --- ┆ --- ┆ --- ┆ --- │ │ datetime[μs] ┆ str ┆ f64 ┆ f64 ┆ bool ┆ f64 ┆ f64 │ ╞═════════════════════╪══════╪═══════╪═══════╪═══════╪══════════╪════════════╡ │ 2026-01-01 09:30:00 ┆ AAA ┆ 1.0 ┆ 10.0 ┆ false ┆ 11.0 ┆ 9.0 │ │ 2026-01-01 09:31:00 ┆ BBB ┆ 0.5 ┆ 10.2 ┆ false ┆ 11.22 ┆ 9.18 │ │ 2026-01-01 09:32:00 ┆ CCC ┆ 0.333 ┆ 10.1 ┆ true ┆ 11.11 ┆ 9.09 │ │ 2026-01-01 09:33:00 ┆ AAA ┆ 0.25 ┆ 10.5 ┆ false ┆ 11.55 ┆ 9.45 │ │ 2026-01-01 09:34:00 ┆ BBB ┆ 1.0 ┆ 10.7 ┆ false ┆ 11.77 ┆ 9.63 │ │ … ┆ … ┆ … ┆ … ┆ … ┆ … ┆ … │ │ 2026-01-01 09:45:00 ┆ AAA ┆ 0.25 ┆ 12.4 ┆ false ┆ 13.64 ┆ 11.16 │ │ 2026-01-01 09:46:00 ┆ BBB ┆ 1.0 ┆ 12.2 ┆ false ┆ 13.42 ┆ 10.98 │ │ 2026-01-01 09:47:00 ┆ CCC ┆ 0.5 ┆ 12.7 ┆ false ┆ 13.97 ┆ 11.43 │ │ 2026-01-01 09:48:00 ┆ AAA ┆ 0.333 ┆ 12.9 ┆ true ┆ 14.19 ┆ 11.61 │ │ 2026-01-01 09:49:00 ┆ BBB ┆ 0.25 ┆ 13.1 ┆ false ┆ 14.41 ┆ 11.79 │ └─────────────────────┴──────┴───────┴───────┴───────┴──────────┴────────────┘
打印输出
shape: (4, 2)
┌──────────┬─────────────────────────────────┐
│ 项目 ┆ 内容 │
│ --- ┆ --- │
│ str ┆ str │
╞══════════╪═════════════════════════════════╡
│ 调用 ┆ col("datetime", "code", "w", "… │
│ 状态 ┆ 未执行成数据表 │
│ 错误类型 ┆ AttributeError │
│ 错误信息 ┆ 'Expr' object has no attribute… │
└──────────┴─────────────────────────────────┘调用
col("datetime", "code", "w", "price", "flag", "limit_up", "limit_down").batch_stock.bt_intraday(100.0, 0.0003)| 参数 | 类型 | 默认值 | 说明 |
|---|---|---|---|
init_money | float | 100.0 | 位置参数 |
fee_rate | float | Params | 0.0003 | 位置参数 |
源码参数说明
init_money:
初始资金(仅第一期使用)。
fee_rate:
手续费率,支持 `float` 或 `Params`。
完整代码
这个算子页使用接口示例:不伪造计算结果。需要真实上下文、多输入源、Monitor session、UDF 回调、策略状态,或当前底层实现之后再执行。
展开可复制完整代码
import datetime as dt
import polars as pl
import qust as qs
from qust import col, pms
data = pl.DataFrame(
{
"datetime": ['2026-01-01 09:30:00', '2026-01-01 09:31:00', '2026-01-01 09:32:00', '2026-01-01 09:33:00', '2026-01-01 09:34:00', '2026-01-01 09:35:00', '2026-01-01 09:36:00', '2026-01-01 09:37:00', '2026-01-01 09:38:00', '2026-01-01 09:39:00', '2026-01-01 09:40:00', '2026-01-01 09:41:00', '2026-01-01 09:42:00', '2026-01-01 09:43:00', '2026-01-01 09:44:00', '2026-01-01 09:45:00', '2026-01-01 09:46:00', '2026-01-01 09:47:00', '2026-01-01 09:48:00', '2026-01-01 09:49:00'],
"code": ['AAA', 'BBB', 'CCC', 'AAA', 'BBB', 'AAA', 'BBB', 'CCC', 'AAA', 'BBB', 'AAA', 'BBB', 'CCC', 'AAA', 'BBB', 'AAA', 'BBB', 'CCC', 'AAA', 'BBB'],
"w": [1.0, 0.5, 0.333, 0.25, 1.0, 0.5, 0.333, 0.25, 1.0, 0.5, 0.333, 0.25, 1.0, 0.5, 0.333, 0.25, 1.0, 0.5, 0.333, 0.25],
"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],
"flag": [False, False, True, False, False, True, False, False, False, True, False, False, False, False, True, False, False, False, True, False],
"limit_up": [11.0, 11.22, 11.11, 11.55, 11.77, 11.44, 11.99, 12.21, 11.88, 12.43, 12.76, 12.54, 13.09, 13.31, 12.98, 13.64, 13.42, 13.97, 14.19, 14.41],
"limit_down": [9.0, 9.18, 9.09, 9.45, 9.63, 9.36, 9.81, 9.99, 9.72, 10.17, 10.44, 10.26, 10.71, 10.89, 10.62, 11.16, 10.98, 11.43, 11.61, 11.79],
}
).with_columns(pl.col("datetime").str.to_datetime())
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('batch_stock.bt_intraday')
print("场景:")
print('股票批处理:面向股票批内回测和交易状态的组合输入。')
print("模式:")
print('接口示例:只构造表达式或对象,不伪造计算结果。')
print("输入列:")
print('datetime, code, w, price, flag, limit_up, limit_down')
print("调用:")
print('col("datetime", "code", "w", "price", "flag", "limit_up", "limit_down").batch_stock.bt_intraday(100.0, 0.0003)')
print("输入数据模板:")
print(data)
try:
result = col("datetime", "code", "w", "price", "flag", "limit_up", "limit_down").batch_stock.bt_intraday(100.0, 0.0003)
except BaseException as err:
out = pl.DataFrame({
"项目": ["调用", "状态", "错误类型", "错误信息"],
"内容": ['col("datetime", "code", "w", "price", "flag", "limit_up", "limit_down").batch_stock.bt_intraday(100.0, 0.0003)', "未执行成数据表", type(err).__name__, str(err)[:200]],
})
print("输出:")
print(out)
else:
out = pl.DataFrame({
"项目": ["调用", "返回类型", "状态", "怎么得到业务表"],
"内容": ['col("datetime", "code", "w", "price", "flag", "limit_up", "limit_down").batch_stock.bt_intraday(100.0, 0.0003)', 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。 |
注意事项
- 先确认输入列名、顺序、类型和本页一致。
- 输出列名不符合业务语义时,显式追加
.alias(...)。 - 窗口和分组类算子要确认
rolling/expanding/over/batch的链式层级。
来源
| 项目 | 位置 |
|---|---|
| 源码文件 | batch_stock.py |
| 类/对象 | BatchStock |