haskell的“哪里”在python的列表理解中

Question

如果lst是一个ints列表，我希望

[dif for i in range(1, len(lst)) where dif = lst[i]-lst[i-1] if dif < 5]

但是显然，绑定这样的变量在语法上是不正确的(除了where是从Haskell借来的)。我能做到

[dif for i in range(1, len(lst)) for dif in (lst[i]-lst[i-1],) if dif < 5]

使dif在长度内运行-一次可迭代。另一种选择：

[lst[i]-lst[i-1] for i in range(1, len(lst)) if lst[i]-lst[i-1] < 5]

此外，我可以定义一个中间列表与差异，然后过滤它。但这两种选择都不是很好。有我想要的好货吗？

Answer 1

为了提高效率，可以使用带有生成器的嵌套理解：

[dif for dif in (lst[i]-lst[i-1] for i in range(1, len(lst))) if dif < 5]

Answer 2

我不认为这里有一条很棒的1条线。我想，因为它是一个列表，所以您可以使用切片：

[a-b for a, b in zip(lst[1:], lst) if a-b < 5]

这不是最有效的，因为lst[1:]创建了一个副本，zip也实现了一个新的列表(在python2.x上)。我们可以使用迭代工具做得更好：

from itertools import izip, islice
[a-b for a, b in izip(islice(lst, 1), lst) if a-b < 5]

(请注意python3.x可以消除izip，所以如果使用python3，只需使用zip)。