Python 熊猫-找到满足每行条件的索引

我试图在数据帧中找到满足一定条件的索引

例如，我们有以下数据帧

并找到这样的索引

argmin(j) df['A'].iloc[j] >= (df['A'].iloc[i] + 3 ) for all i 

因此，结果将由

我使用for循环完成了这项工作，但我相信有更有效的方法来完成这项工作

谢谢你的回复

我的代码是

for i in range(len(df)):
    df['B'].iloc[i] = df[df2['A']>= df2['A'].iloc[i]+1].index[0]

但是，for循环对于大型数据集来说太慢了

试试下面的方法：）

---

欢迎来到StackOverflow。请不要发布数据截图的链接，只需内联包含示例数据即可。这使得人们可以更容易地帮助您，并且可以防止链接在将来失效。创建最小的工作示例，以便每个人都可以运行它并进行修改。谢谢！这对我帮助很大！

import pandas as pd
import numpy as np

df = pd.DataFrame({'A': [1,3,5,8,10,12]})
b = pd.DataFrame(df.values - (df['A'].values + 3), index=df.index)
df['B'] = b.where(b >= 0).idxmin()
df