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逆转一下。
分类:
Pandas
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 阿里最新开源QwQ-32B,效果媲美deepseek-r1满血版,部署成本又又又降低了!
· 开源Multi-agent AI智能体框架aevatar.ai,欢迎大家贡献代码
· Manus重磅发布:全球首款通用AI代理技术深度解析与实战指南
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· AI技术革命,工作效率10个最佳AI工具