停止列表选择？ [复制]

Question

假设我有一个元组的顺序列表：

s = [(0,-1), (1,0), (2,-1), (3,0), (4,0), (5,-1), (6,0), (7,-1)]

给定一个参数X，我想选择第一个元素等于或大于X 的所有元组，但不包括第一个元组的第二个元素为-1。

比如X = 3，我想选择列表[(3,0), (4,0)]

我的一个想法是： 用

获取截止键

E = min (x [0] for x in s if (x [0] >= X) and (x [1] == -1) )

然后用X和E之间的键选择元素：

R = [x for x in s if X <= x [0] < E]

这给了我在 R 中想要的东西，但它似乎真的效率低下，涉及两个表扫描。我可以在 for 循环中执行此操作，丢弃键太小的元组，并在遇到第一个阻塞元组时 break。但是与列表选择相比，像狗一样奔跑。

有没有一种超级高效的python-esque (2.7) 方式来做到这一点？

Answer 1

在生成器表达式中实现takewhile 有点麻烦：

def stop(): raise StopIteration

result = (
    stop() if item[1] == -1 else
    item
    for item in s
    if item[0] >= X
)

或者用不同的措辞：

def until(cond):
    if cond: raise StopIteration
    return True

result = (
    item for item in s
    if item[0] >= X
    and until(item[1] == -1)
)

Answer 2

您可以简单地将列表中的元组过滤为生成器表达式，然后当您获得第二个元素为-1 的第一个元组时，您可以停止从生成器表达式中获取值，就像这样

>>> s = [(0,-1), (1,0), (2,-1), (3,0), (4,0), (5,-1), (6,0), (7,-1)]
>>> from itertools import takewhile
>>> X = 3
>>> list(takewhile(lambda x: x[1] != -1, (item for item in s if item[0] >= X)))
[(3, 0), (4, 0)]

这里，生成器表达式(item for item in s if item[0] >= X) 将根据需要一个接一个地给出值（它们不是一次全部生成的，所以我们在这里节省内存）大于或等于X。

然后，我们从该生成器表达式中获取值，直到我们找到第二个元素不等于 -1 的元组与 itertools.takewhile。