python - Python は Ruby の #each_cons に相当しますか?

Question

Rubyのに相当するPythonicはあり#each_consますか?

Ruby では、次のことができます。

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

Answer 1

あるとは思いません。組み込みのモジュールを調べましたitertools。これは、私が期待する場所です。ただし、簡単に作成できます。

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

Answer 2

そのようなもののために、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 3

リストの私のソリューション（Python2）：

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

編集: Python 3itertools.izipではなくなったので、plain を使用しますzip:

def each_cons(xs, n):
    return zip(*(xs[i:] for i in range(n)))

Answer 4

更新：以下の私の答えを気にせず、ただ使用toolz.itertoolz.sliding_window()してください-それは正しいことをします.

---

each_consシーケンス/ジェネレーターの長さが不十分な場合に Ruby の動作を保持する、真に遅延した実装の場合:

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 の引数で使用されるタプルのアンパッキングは、反復したいシーケンスではなく、n結果のサイズitertools.tee(xs, n)(つまり、「ウィンドウ サイズ」) のタプルに適用されることに注意してください。

Answer 5

Pythonは確かにこれを行うことができます。あまり熱心にやりたくない場合は、itertool の islice と izip を使用してください。また、通常のスライスはコピーを作成することを覚えておくことが重要です。メモリ使用量が重要な場合は、同等の itertool も考慮する必要があります。

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

Answer 6

簡単なワンライナー：

a = [1, 2, 3, 4]

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