打破清单理解

Question

如何根据条件中断列表理解,例如何时412找到数字？

码:

numbers = [951, 402, 984, 651, 360, 69, 408, 319, 601, 485, 980, 507, 725, 547, 544,
           615, 83, 165, 141, 501, 263, 617, 865, 575, 219, 390, 984, 592, 236, 105, 942, 941,
           386, 462, 47, 418, 907, 344, 236, 375, 823, 566, 597, 978, 328, 615, 953, 345, 399,
           162, 758, 219, 918, 237, 412, 566, 826, 248, 866, 950, 626, 949, 687, 217, 815, 67,
           104, 58, 512, 24, 892, 894, 767, 553, 81, 379, 843, 831, 445, 742, 717, 958, 609, 842,
           451, 688, 753, 854, 685, 93, 857, 440, 380, 126, 721, 328, 753, 470, 743, 527]

even = [n for n in numbers if 0 == n % 2]

所以在功能上,你可以推断这应该做的事情:

even = [n for n in numbers if 0 == n % 2 and break if n == 412]

我真的更喜欢:

• 单行
• 没有其他花哨的库,如itertools,"纯python",如果可能的话(读:解决方案不应该使用任何import语句或类似)

Answer 1

使用函数来引发StopIteration并list捕获它:

>>> def end_of_loop():
...     raise StopIteration
... 
>>> even = list(end_of_loop() if n == 412 else n for n in numbers if 0 == n % 2)
>>> print(even)
[402, 984, 360, 408, 980, 544, 390, 984, 592, 236, 942, 386, 462, 418, 344, 236, 566, 978, 328, 162, 758, 918]

对于那些抱怨它不是一个单行:

even = list(next(iter(())) if n == 412 else n for n in numbers if 0 == n % 2)

对于那些抱怨的人来说,它是hackish,不应该用在生产代码中:嗯,你是对的.当然.

Answer 2

您可以将生成器表达式与itertools.takewhile():

even_numbers = (n for n in numbers if not n % 2)
list(itertools.takewhile(lambda x: x != 412, even_numbers))

编辑:我刚刚注意到要求不使用任何imports.好吧,无论如何,我在这里留下这个答案.

Answer 3

如果412肯定会在列表中,你可以使用这个:

even = [n for n in numbers[:numbers.index(412)] if not n % 2]

如果要在结果中包含412,则仅numbers[:numbers.index(412)+1]用于切片.

请注意,由于切片,这将比itertools或for循环解决方案效率低(至少在内存方面).

Answer 4

even = [n for n in numbers[:None if 412 not in numbers else numbers.index(412)] if not n % 2] 

刚刚上面的FJ代码并添加了一个三元组来检查412是否在列表中.仍然是"一个班轮",即使412不在列表中也能工作.

Answer 5

我知道这是一篇非常古老的帖子,但是由于OP询问break内部使用list-comprehension,我也在寻找类似的东西,我想我会在这里发布我的发现以供将来参考.

在调查break时,我看到鲜为人知的特性来了iter作为iter(callable, sentinel)其返回一个迭代器,"休息"迭代一次赎回function价值等于sentinel价值.

>>> help(iter)
Help on built-in function iter in module __builtin__:

iter(...)
    iter(collection) -> iterator
    iter(callable, sentinel) -> iterator

    Get an iterator from an object.  In the first form, the argument must
    supply its own iterator, or be a sequence.
    In the second form, the callable is called until it returns the sentinel.

这里棘手的部分是定义一个适合给定问题的函数.在这种情况下,我们首先需要转换给定list的numbers一个来iterator使用x = iter(numbers)外部变量成饲料lambda的功能.

接下来,我们的可调用函数只是调用迭代器来吐出下一个值.然后迭代器与我们的标记值(在这种情况下为412)进行比较,并在达到该值时"中断".

print [i for i in iter(lambda x=iter(numbers): next(x),412) if i %2 == 0]

>>> 
[402, 984, 360, 408, 980, 544, 390, 984, 592, 236, 942, 386, 462, 418,  
 344, 236, 566, 978, 328, 162, 758, 918]