Pandas

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

Pandas - 找到每一行满足条件的索引

Pandas - find the index satisfying conditions of each row

python

sklearn-pandas