我有以下氨纶:
df = pl.DataFrame({
"Column A": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
"Column B": [2, 3, 1, 4, 1, 7, 3, 2, 12, 0]
})
我想创建一个新的列C，以行为单位保存当前行的B值与列B中的下一个值之间的距离，该距离大于或等于B +50%.
最终结果应该是这样的:
df = pl.DataFrame({
"Column A": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
"Column B": [2, 3, 1, 4, 1, 7, 3, 2, 12, 0],
"Column C": [1, 4, 1, 2, 1, 3, 2, 1, None, None]
})
如何使用Polars有效地实现这一目标，尤其是因为我正在使用大型数据框架？
好吧，首先我应该说--这个看起来需要在多列上加入不等式，根据我发现纯极点并不适合它.用join_asof个可能可以做到这一点，但我无法让它漂亮.
我可能会使用duckdb integration with polars来实现结果:
import duckdb
duckdb.sql("""
select
d."Column A",
d."Column B",
(
select tt."Column A"
from df as tt
where tt."Column A" > d."Column A" and tt."Column B" >= d."Column B" * 1.5
order by tt."Column A" asc
limit 1
) - d."Column A" as "Column C"
from df as d
""").pl()
┌──────────┬──────────┬──────────┐
│ Column A ┆ Column B ┆ Column C │
│ --- ┆ --- ┆ --- │
│ i64 ┆ i64 ┆ i64 │
╞══════════╪══════════╪══════════╡
│ 1 ┆ 2 ┆ 1 │
│ 2 ┆ 3 ┆ 4 │
│ 3 ┆ 1 ┆ 1 │
│ 4 ┆ 4 ┆ 2 │
│ 5 ┆ 1 ┆ 1 │
│ 6 ┆ 7 ┆ 3 │
│ 7 ┆ 3 ┆ 2 │
│ 8 ┆ 2 ┆ 1 │
│ 9 ┆ 12 ┆ null │
│ 10 ┆ 0 ┆ null │
└──────────┴──────────┴──────────┘