python - 找到给定数字上方和下方最接近的元素

Question

myArr = array([4,1,88,44,3])
myNumber = 25
FindClosest(myArr, myNumber)
...
4, 44

有什么方法可以找到列表中最接近给定数字的 2 个数字，其中一个较大，另一个较小？

我可以通过以下方式找到最接近的数字:

min(myArr.tolist(), key=lambda x:abs(x-myNumber))

Answer 1

排序不是必需的，它使这个时间复杂度为 O(n logn)，而它应该只是 O(n)。

我相信这就是您正在寻找的，利用 numpy 数组索引:

>>> # the smallest element of myArr greater than myNumber
>>> myArr[myArr > myNumber].min()  
44

>>> # the largest element of myArr less than myNumber
>>> myArr[myArr < myNumber].max()
4