通過兩個熊貓數據框中的值查找閉合行而無需迭代

Question

假設我有兩個數據幀df1和df2，我想執行以下操作：

for i, row in df1.iterrows():
    lines = df2[(df2['A'] - row['A']).abs() < threshold]
    arr.extend(lines.values)

這種方法效率極低，如何在不進行行迭代的情況下做到這一點？ 我知道我可以使用Cython改善迭代，但是我想避免這樣做。

Answer 1

如果df1和df2共享一個索引，則可以創建一個布爾掩碼，然后通過它創建df2索引。

mask = (df2.A - df1.A.abs()) < threshold
arr = df2[mask]