在元组列表中查找索引

Question

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

我这样做的代码如下:

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

但是,如果索引位于列表seg的第一个第二个段中,我也想返回.

例如,对于index = 876返回1和index = 1015返回3.

我怎么能这样做？

Answer 1

你可以使用enumerate+ nextwith generator expression:

>>> 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的提示.(它可以在给定索引不在元组表示的任何范围内的情况下使用)