python - 比较两个不同大小的一维数组中的元素，具有公差

Question

<分区>

Answer 1

您可以通过使用稍微更内置的 numpy 解决方案来改进嵌套循环:

import numpy as np

A = np.array([0, 0.3141, 1.234, 4.1341, -34.112])
B = np.array([0.3142, 2.234, 4.1340, -34.113, 40])
res = {i for i in A if np.isclose(B, i, 0.1).any()}
print(res)

输出:

{0.3141, 4.1341, -34.112}

时间:

In [2]: %timeit {i for i in A if np.isclose(B, i, 0.1).any()}
276 µs ± 16.2 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)

In [3]: %timeit {x for x in A for y in B if np.isclose(x, y, atol=0.1)}
1.08 ms ± 12 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)