Python等效於Ruby的#each_cons？

Question

Pythonic是否與Ruby的#each_cons相同？

在Ruby中，你可以這樣做：

array = [1,2,3,4] 
array.each_cons(2).to_a 
=> [[1,2],[2,3],[3,4]] 

Answer 1

對於這樣的事情，itertools是你應該尋找在模塊：

from itertools import tee, izip 

def pairwise(iterable): 
    "s -> (s0,s1), (s1,s2), (s2, s3), ..." 
    a, b = tee(iterable) 
    next(b, None) 
    return izip(a, b) 

然後：

>>> list(pairwise([1, 2, 3, 4])) 
[(1, 2), (2, 3), (3, 4)] 

對於更普遍的解決方案，請考慮：

def split_subsequences(iterable, length=2, overlap=0): 
    it = iter(iterable) 
    results = list(itertools.islice(it, length)) 
    while len(results) == length: 
     yield results 
     results = results[length - overlap:] 
     results.extend(itertools.islice(it, length - overlap)) 
    if results: 
     yield results 

這允許任意長度的子序列和任意重疊。用法：

>> list(split_subsequences([1, 2, 3, 4], length=2)) 
[[1, 2], [3, 4]] 
>> list(split_subsequences([1, 2, 3, 4], length=2, overlap=1)) 
[[1, 2], [2, 3], [3, 4], [4]] 

Answer 2

一個快速班輪：

a = [1, 2, 3, 4] 

out = [a[i:i + 2] for i in range(len(a) - 1)] 

Answer 3

我不認爲這是一個，我通過看內置模塊itertools，這是我希望它是。你可以簡單地創建一個：

def each_cons(x, size): 
    return [x[i:i+size] for i in range(len(x)-size+1)] 

Answer 4

Python肯定可以做到這一點。如果你不想這麼熱切地使用itertool的islice和izip。此外，重要的是要記住正常的切片將創建一個副本，所以如果內存使用很重要，你還應該考慮itertool等價物。

each_cons = lambda l: zip(l[:-1], l[1:])

Answer 5

我對名單的解決方案：

import itertools 
def each_cons(xs, n): 
    return itertools.izip(*(xs[i:] for i in xrange(n))) 

Answer 6

UPDATE：沒關係我的回答如下，只是用toolz.itertoolz.sliding_window() - 它會做正確的事情。

---

對於一個真正的懶惰實現，它保留了Ruby的each_cons的行爲時，順序/發電機具有足夠的長度：

import itertools 
def each_cons(sequence, n): 
    return itertools.izip(*(itertools.islice(g, i, None) 
          for i, g in 
          enumerate(itertools.tee(sequence, n)))) 

例子：

>>> print(list(each_cons(xrange(5), 2))) 
[(0, 1), (1, 2), (2, 3), (3, 4)] 
>>> print(list(each_cons(xrange(5), 5))) 
[(0, 1, 2, 3, 4)] 
>>> print(list(each_cons(xrange(5), 6))) 
[] 
>>> print(list(each_cons((a for a in xrange(5)), 2))) 
[(0, 1), (1, 2), (2, 3), (3, 4)] 

注意拆包使用的元組對於izip的參數應用於由itertools.tee(xs, n)（即「窗口大小」）產生的大小爲n的元組，而不是我們想要迭代的序列。

Answer 7

同埃利亞斯的代碼，但適用於Python 2和3：

try: 
    from itertools import izip # python 2 
except ImportError: 
    from builtins import zip as izip # python 3 

from itertools import islice, tee 

def each_cons(sequence, n): 
    return izip(
     *(
      islice(g, i, None) 
      for i, g in 
      enumerate(tee(sequence, n)) 
     ) 
    )