Подтвердить что ты не робот

Итерации по последовательности python в кратных n?

Как обрабатывать элементы последовательности в пакетах, идиоматически?

Например, с последовательностью "abcdef" и размером партии 2, я хотел бы сделать что-то вроде следующего:

for x, y in "abcdef":
    print "%s%s\n" % (x, y)
ab
cd
ef

Конечно, это не работает, потому что он ожидает, что из списка будет один элемент, который сам содержит 2 элемента.

Что такое хороший, короткий, чистый, pythonic способ обработки следующих n элементов списка в пакетном режиме или подстроки длины n из большей строки (две аналогичные проблемы)?

4b9b3361

Ответ 1

Я уверен, что кто-то придумает еще несколько "Pythonic", но как насчет:

for y in range(0, len(x), 2):
    print "%s%s" % (x[y], x[y+1])

Обратите внимание, что это будет работать, только если вы знаете, что len(x) % 2 == 0;

Ответ 2

Функция генератора была бы аккуратной:

def batch_gen(data, batch_size):
    for i in range(0, len(data), batch_size):
            yield data[i:i+batch_size]

Пример использования:

a = "abcdef"
for i in batch_gen(a, 2): print i

печатает:

ab
cd
ef

Ответ 3

У меня есть альтернативный подход, который работает для итераций, которые не имеют известной длины.

   
def groupsgen(seq, size):
    it = iter(seq)
    while True:
        values = ()        
        for n in xrange(size):
            values += (it.next(),)        
        yield values    

Он работает путем повторения последовательности (или другого итератора) в группах по размеру, собирая значения в кортеже. В конце каждой группы это дает кортеж.

Когда итератор исчерпал значения, он выдает исключение StopIteration, которое затем распространяется вверх, указывая, что у группgen вне значений.

Он предполагает, что значения входят в набор размеров (наборы из 2, 3 и т.д.). Если нет, любые оставшиеся значения просто отбрасываются.

Ответ 4

Не забывайте о функции zip():

a = 'abcdef'
for x,y in zip(a[::2], a[1::2]):
  print '%s%s' % (x,y)

Ответ 5

но более общий способ был бы (вдохновлен этим ответом):

for i in zip(*(seq[i::size] for i in range(size))):
    print(i)                            # tuple of individual values

Ответ 6

И тогда всегда существует документация.

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

def grouper(n, iterable, padvalue=None):
    "grouper(3, 'abcdefg', 'x') --> ('a','b','c'), ('d','e','f'), ('g','x','x')"
    return izip(*[chain(iterable, repeat(padvalue, n-1))]*n)

Примечание: они производят кортежи вместо подстрок, когда в качестве ввода задана последовательность строк.

Ответ 7

>>> a = "abcdef"
>>> size = 2
>>> [a[x:x+size] for x in range(0, len(a), size)]
['ab', 'cd', 'ef']

.. или, не как понимание списка:

a = "abcdef"
size = 2
output = []
for x in range(0, len(a), size):
    output.append(a[x:x+size])

Или, как генератор, который был бы лучше, если бы он использовался несколько раз (для одноразовой вещи понимание списка, вероятно, "лучше" ):

def chunker(thelist, segsize):
    for x in range(0, len(thelist), segsize):
            yield thelist[x:x+segsize]

.. и его использование:

>>> for seg in chunker(a, 2):
...     print seg
... 
ab
cd
ef

Ответ 8

вы можете создать следующий генератор

def chunks(seq, size):
    a = range(0, len(seq), size)
    b = range(size, len(seq) + 1, size)
    for i, j in zip(a, b):
        yield seq[i:j]

и используйте его следующим образом:

for i in chunks('abcdef', 2):
    print(i)

Ответ 9

Из документов more_itertools: more_itertools.chunked()

more_itertools.chunked(iterable, n)

Разбейте итерацию на списки заданной длины:

>>> list(chunked([1, 2, 3, 4, 5, 6, 7], 3))
[[1, 2, 3], [4, 5, 6], [7]]

Если длина повторяемого элемента не делится поровну на n, последний возвращенный список будет короче.

Ответ 10


s = 'abcdefgh'
for e in (s[i:i+2] for i in range(0,len(s),2)):
  print(e)

Ответ 11

itertools doc имеет рецепт для этого:

from itertools import izip_longest

def grouper(iterable, n, fillvalue=None):
    "Collect data into fixed-length chunks or blocks"
    # grouper('ABCDEFG', 3, 'x') --> ABC DEF Gxx
    args = [iter(iterable)] * n
    return izip_longest(fillvalue=fillvalue, *args)

Использование:

>>> l = [1,2,3,4,5,6,7,8,9]
>>> [z for z in grouper(l, 3)]
[(1, 2, 3), (4, 5, 6), (7, 8, 9)]

Ответ 12

Одно решение, хотя я призываю кого-то сделать лучше; -)

a = 'abcdef'
b = [[a[i-1], a[i]] for i in range(1, len(a), 2)]

for x, y in b:
  print "%s%s\n" % (x, y)

Ответ 13

Как насчет itertools?

from itertools import islice, groupby

def chunks_islice(seq, size):
    while True:
        aux = list(islice(seq, 0, size))
        if not aux: break
        yield "".join(aux)

def chunks_groupby(seq, size):
    for k, chunk in groupby(enumerate(seq), lambda x: x[0] / size):
        yield "".join([i[1] for i in chunk])

Ответ 14

За исключением двух ответов, я видел много преждевременных материализаций по партиям и подписи (что не работает для всех итераторов). Поэтому я придумал эту альтернативу:

def iter_x_and_n(iterable, x, n):
    yield x
    try:
        for _ in range(n):
            yield next(iterable)
    except StopIteration:
        pass

def batched(iterable, n):
    if n<1: raise ValueError("Can not create batches of size %d, number must be strictly positive" % n)
    iterable = iter(iterable)
    try:
        for x in iterable:
            yield iter_x_and_n(iterable, x, n-1)
    except StopIteration:
        pass

Мне кажется, что для этого нет ни однострочного, ни малоличного решения (насколько я знаю). Основная проблема заключается в том, что как внешний генератор, так и внутренний генератор должны правильно обрабатывать StopIteration. Внешний генератор должен давать только что-то, если осталось хотя бы один элемент. Интуитивно понятный способ проверить это - выполнить следующий (...) и уловить StopIteration.

Ответ 15

Учитывая

from __future__ import print_function                      # python 2.x

seq = "abcdef"
n = 2

код

while seq:
    print("{}".format(seq[:n]), end="\n")
    seq = seq[n:]

Выход

ab
cd
ef

Ответ 16

Вот решение, которое дает серию итераторов, каждый из которых выполняет итерацию по n элементам.

def groupiter(thing, n):
    def countiter(nextthing, thingiter, n):
        yield nextthing
        for _ in range(n - 1):
            yield next(thingiter)
    thingiter = iter(thing)
    while True:
        try:
            nextthing = next(thingiter)
        except StopIteration:
            return None
        yield countiter(nextthing, thingiter, n)

Я использую это следующим образом:

table = list(range(256))
for group in groupiter(table, 16):
    print(', '.join('0x{:02X}'.format(x) for x in group))