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逆转一下。

标签：df,random,DataFrame,argmax,cond,np,上次,符合条件
来源： https://www.cnblogs.com/yuyanc/p/16412945.html