小python怪癖：查找列表边界内第一个和最后一个数字的索引

Question

我遇到了一个烦人的小问题。我的问题是这样的：

我有一系列介于 0 和 1 之间的数字： [0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9]

还有两个边界，比如 0.25 和 0.75。

我需要一种快速而漂亮的方法来找到系列中第一个数字和最后一个数字的索引，它们在边界内，在本例中为 (2, 6)

到目前为止，我只提出了一种使用 for 循环和 break 命令的笨拙方法。

提前感谢您的帮助！

Answer 1

如果你可以使用 numpy：

import numpy as np
data = np.array([0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9])
max_b = .75
min_b = .25
wh = np.where((data < max_b)*(data > min_b))[0]
left, right =  wh[0], wh[-1] + 1

或者简单地说（感谢 dougal）：

left, right = np.searchsorted(data, [min_b, max_b])

如果你不能：

import bisect

data = [0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9]
max_b = .75
min_b = .25
left = bisect.bisect_left(data, min_b)
right = bisect.bisect_right(data, max_b)

右侧加或减 1，具体取决于您是希望 data[right] 在集合中，还是 data[left:right] 为您提供集合。

Answer 2

如果您的数字序列始终排序，您可以使用the bisect module 对端点执行binary search：

>>> a = [.1, .2, .3, .4, .5, .6, .7, .8, .9]
>>> import bisect
>>> bisect.bisect_left(a, 0.25)
2
>>> bisect.bisect_right(a, 0.75) - 1
6

bisect_left(a, x) 返回位置p 使得a[:p] 的每个元素都小于x，并且a[p:] 的每个元素都大于或等于x；这正是您想要的下限。

bisect_right 返回位置p 使得a[:p] 的每个元素都小于或等于x，并且a[p:] 都大于x。所以右边界需要减一才能得到最大位置<= x。