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

Question

我试图在pandas DataFrame中找到满足某些条件的索引。

例如，我们有以下数据框

找到索引

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循环对于大型数据集来说太慢了。

Answer 1

尝试以下方法：）

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