Возможный дубликат:
Как вы разбиваете список на четные фрагменты в Python?
Я удивлен, что не смог найти "пакетную" функцию, которая принимала бы как входную итерабельность и возвращала итерабельность итераций.
Например:
for i in batch(range(0,10), 1): print i
[0]
[1]
...
[9]
или
for i in batch(range(0,10), 3): print i
[0,1,2]
[3,4,5]
[6,7,8]
[9]
Теперь я написал то, что считал довольно простым генератором:
def batch(iterable, n = 1):
current_batch = []
for item in iterable:
current_batch.append(item)
if len(current_batch) == n:
yield current_batch
current_batch = []
if current_batch:
yield current_batch
Но вышеизложенное не дает мне того, что я ожидал бы:
for x in batch(range(0,10),3): print x
[0]
[0, 1]
[0, 1, 2]
[3]
[3, 4]
[3, 4, 5]
[6]
[6, 7]
[6, 7, 8]
[9]
Итак, я что-то пропустил, и это, вероятно, показывает мое полное отсутствие понимания генераторов питона. Кто-нибудь хотел бы указать мне в правильном направлении?
[Edit: В конце концов я понял, что описанное выше поведение происходит только тогда, когда я запускаю его внутри ipython, а не сам python]