DataFrame计算上次符合条件到现在的位置

想实现类似通达信BARSLAST 上次条件成立到现在的位置

得到大佬帮助,帮我写了以下代码

import pandas as pd
import numpy as np

df = pd.DataFrame(np.random.random((10,4)),
                  columns = ['A', 'B', 'cond', 'D'])
# 让 cond 列变成 True/False
df.cond = df.cond > 0.5

# 最近一次 True 到当前 row 距离?
df['barslast']=df['cond'].rolling(len(df), min_periods = 1).apply(lambda x: np.argmax(x[::-1]))
print(df)

argmax为何要把数组逆转?是因为argmax看的是第一次max所在的index,我们是倒着往上找,所以这里把x逆转一下。

posted @   C羽言  阅读(102)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 阿里最新开源QwQ-32B,效果媲美deepseek-r1满血版,部署成本又又又降低了!
· 开源Multi-agent AI智能体框架aevatar.ai,欢迎大家贡献代码
· Manus重磅发布:全球首款通用AI代理技术深度解析与实战指南
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· AI技术革命,工作效率10个最佳AI工具
点击右上角即可分享
微信分享提示