查找与python

时间:2017-12-14 12:30:21

标签: python arrays performance numpy

我正在研究一个问题,我很快就需要找到最接近另一个数组(array1)中的值的数组(array2)的索引,并为array1的每个值提供索引。 array1和array2的大小永远不会相同。

我目前的职能如下:

def findClosestArrayIndices(array1, array2):
    indices = []
    for value in array1:
        indices.append((np.abs(array2 - value)).argmin())
    return np.array(indices)

这可以按预期工作,但由于for循环遍历array1中的每个值,因此速度很慢,尤其是当array1变大时。是否有更多的pythonic / numpy方法来重写此函数来切除循环并加快进程?

谢谢

0 个答案:

没有答案