разделить генератор / повторяемый каждые n элементов в Python (splitEvery)

Почему бы не сделать это так? Выглядит почти как ваша splitEvery_2 функция.

def splitEveryN(n, it):
    return [it[i:i+n] for i in range(0, len(it), n)]

На самом деле это только убирает ненужный интервал шага из среза в вашем решении. :)

Однострочное, встроенное решение для этого (поддерживает v2 / v3, итераторы, использует стандартную библиотеку и понимание одного генератора):

import itertools
def split_groups(iter_in, group_size):
     return ((x for _, x in item) for _, item in itertools.groupby(enumerate(iter_in), key=lambda x: x[0] // group_size))

Вот как вы работаете со списком и итератором:

def isList(L): # Implement it somehow - returns True or false
...
return (list, lambda x:x)[int(islist(L))](result)

more_itertools имеет функцию chunked :

import more_itertools as mit


list(mit.chunked(range(9), 5))
# [[0, 1, 2, 3, 4], [5, 6, 7, 8]]

Вот быстрая однострочная версия. Как и у Хаскелла, он ленивый.

from itertools import islice, takewhile, repeat
split_every = (lambda n, it:
    takewhile(bool, (list(islice(it, n)) for _ in repeat(None))))

Для этого необходимо использовать iter до вызова split_every.

Пример:

list(split_every(5, iter(xrange(9))))
[[0, 1, 2, 3, 4], [5, 6, 7, 8]]

Хотя это и не одна строка, в приведенной ниже версии не требуется, чтобы вы называли iter, что может быть распространенной ошибкой.

from itertools import islice, takewhile, repeat

def split_every(n, iterable):
    """
    Slice an iterable into chunks of n elements
    :type n: int
    :type iterable: Iterable
    :rtype: Iterator
    """
    iterator = iter(iterable)
    return takewhile(bool, (list(islice(iterator, n)) for _ in repeat(None)))

(Спасибо @ eli-korvigo за улучшения.)

Я думаю, что эти вопросы почти равны

. Если немного подрезать последние, я думаю, что хорошим решением для случая генератора было бы:

from itertools import *
def iter_grouper(n, iterable):
    it = iter(iterable)
    item = itertools.islice(it, n)
    while item:
        yield item
        item = itertools.islice(it, n)

для объекта, который поддерживает слайсы (списки, строки, кортежи), мы можем сделать:

def slice_grouper(n, sequence):
   return [sequence[i:i+n] for i in range(0, len(sequence), n)]

теперь это просто вопрос выбора правильного метода:

def grouper(n, iter_or_seq):
    if hasattr(iter_or_seq, "__getslice__"):
        return slice_grouper(n, iter_or_seq)
    elif hasattr(iter_or_seq, "__iter__"):
        return iter_grouper(n, iter_or_seq)

Я думаю, вы могли бы немного отполировать: -)

это сделает трюк

from itertools import izip_longest
izip_longest(it[::2], it[1::2])

, где * it * является итеративным

Пример:

izip_longest('abcdef'[::2], 'abcdef'[1::2]) -> ('a', 'b'), ('c', 'd'), ('e', 'f')

Давайте разберем это

'abcdef'[::2] -> 'ace'
'abcdef'[1::2] -> 'bdf'

Как вы можете видеть, последний номер в срезе указывает интервал, который будет использоваться для сбора предметов. Вы можете прочитать больше об использовании расширенных срезов здесь .

Функция zip берет первый элемент из первой итерируемой и объединяет его с первым элементом со второй итерируемой. Затем функция zip делает то же самое для второго и третьего элементов, пока у одного из итераций не заканчиваются значения.

Результатом является итератор. Если вы хотите список, используйте функцию list () для результата.