表达式核心 / expr.py

expr.near

布尔范围扩展算子，找到输入 bool 列中第一次和最后一次 `True`，并向两端扩展。

可执行示例returns: Exprexpr

输入 / 输出

输入

flag

输入项	类型	示例
`flag`	`Boolean`	False

输出

项目	说明
返回类型	`Expr`
输出对象	Expr；执行后得到 Polars DataFrame
输出语义	输出列由算子、alias 或底层实现决定；需要稳定列名时显式使用 alias。
执行方式	用 col(...).runtime() 或 col.with_cols(...).runtime() 创建执行计划后 calc_data。
核心调用	`col("flag").near(1)`

打印输入 / 打印输出

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

打印输入

print(data)

shape: (20, 1)
┌───────┐
│ flag  │
│ ---   │
│ bool  │
╞═══════╡
│ false │
│ false │
│ true  │
│ false │
│ false │
│ …     │
│ false │
│ false │
│ false │
│ true  │
│ false │
└───────┘

打印输出

print(out)

shape: (20, 1)
┌───────┐
│ flag  │
│ ---   │
│ bool  │
╞═══════╡
│ false │
│ true  │
│ true  │
│ false │
│ false │
│ …     │
│ false │
│ false │
│ false │
│ true  │
│ true  │
└───────┘

调用

Python

col("flag").near(1)

参数	类型	默认值	说明
`n`	`int`	`必填`	位置参数
`head`	`bool`	`True`	位置参数
`tail`	`bool`	`True`	位置参数

源码参数说明

- `n`：扩展的行数。
- `head=True`：是否把第一次 `True` 向前扩展 `n` 行。
- `tail=True`：是否把最后一次 `True` 向后扩展 `n` 行。

完整代码

本页完整例子会执行真实的 calc_data 或对象调用。

展开可复制完整代码

Python

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

data = pl.DataFrame(
    {
        "flag": [False, False, True, False, False, True, False, False, False, True, False, False, False, False, True, False, False, False, True, False],
    }
)

print("算子:")
print('expr.near')
print("场景:")
print('表达式核心：在执行计划里组合计算、窗口、过滤或元数据操作。')
print("模式:")
print('可执行示例：构造表达式并运行 calc_data。')
print("输入列:")
print('flag')
print("调用:")
print('col("flag").near(1)')
print("输入数据:")
print(data)
expr = col("flag").near(1)
df = col(expr).runtime()
out = df.calc_data(data)
print("输出:")
print(out)

改成业务代码

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

注意事项

- 输入必须是一列 bool。
- `over(...)` 上下文下，每个 group 单独找自己的第一次和最后一次 `True`，不会跨组扩展。

来源

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