パンダ - 各行の条件を満たすインデックスを見つける

Question

パンダの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] 

ですが、ループのための大規模なデータセットのために遅すぎます。

Answer 1

方法次のtry、:)

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