在元组列表中查找索引

Question

我有一个元组seg = [(874, 893), (964, 985), (1012, 1031)]和一个索引。 我想检查索引是否在那些元组的范围内，例如876是，而870不是。

我的代码如下：

if [x for (x, y) in seg if x <= index <= y]:
   print ("index inside the segment")

但是，我也想返回索引是否在列表段的第一第二...段中。

例如，对于index = 876返回1 ，对于index = 1015返回3 。

我该怎么办？

Answer 1

您可以在生成器表达式中使用enumerate + next ：

>>> seg = [(874, 893), (964, 985), (1012, 1031)]
>>> index = 876
>>> next((i for i, (s,f) in enumerate(seg) if s <= index <= f), None)
0

---

或者，如果您要迭代：

>>> for i in (i for i, (s,f) in enumerate(seg) if s <= index <= f):
...     print("in segment:", i)
... 
in segment: 0

---

感谢@jpp提供有关下next功能默认选项的提示。 （可以在给定索引不在元组表示的任何范围内的情况下使用）

Answer 2

正如其他人指出的那样，您可以使用enumerate()获取索引。 我还认为，如果将元组视为范围，则应将其设置为range 。 然后，这可以非常直观地检查值是否在范围内： value in range 。

import itertools

seg = [(874, 893), (964, 985), (1012, 1031)]
ranges = list(itertools.starmap(range, seg))

def test(value):
  for i, valueRange in enumerate(ranges):
    if value in valueRange:
      return i # + 1 if you want to index from 1 as indicated.
  # You could add some special case handling here if there is no match, like:
  # throw NoSuchRangeException("The given value was not inside any of the ranges.")

print(test(876)) # 0
print(test(1015)) # 1

显然，使用范围会产生一定的成本（如果您使用的是Python 2.x，这会比较大，因为它将生成所有值的实际列表，并且不幸的是， xrange()返回没有实现__contains__()对象）。 如果您在很多地方都在做这种事情，那会更好。

根据情况，您也许可以只用范围结构替换元组结构，而不用做星形图。

Answer 3

假设以下内容：

清单已订购

第一个数字小于第二个数字

没有重叠

index=870
seg = [(874, 893), (964, 985), (1012, 1031)]
for i, t in enumerate(seg):
    if index >= t[0] and index <= t[1]:
        print i