import itertools
with open(fn) as f:
for line, line2 in itertools.izip_longest(f, f, fillvalue=''):
print line, line2
Увы, izip_longest
требует Python 2.6 или выше; 2.5 имеет только izip
, что усечет последнюю строку, если f
имеет нечетное количество строк. Конечно, довольно просто предоставить эквивалентную функциональность в качестве генератора.
Вот более общий итератор-оболочка «N за один раз»:
def natatime(itr, fillvalue=None, n=2):
return itertools.izip_longest(*(iter(itr),)*n, fillvalue=fillvalue)
itertools
, как правило, лучший путь, но, если вы настаивали на его реализации самостоятельно, то:
def natatime_no_itertools(itr, fillvalue=None, n=2):
x = iter(itr)
for item in x:
yield (item,) + tuple(next(x, fillvalue) for _ in xrange(n-1))
В 2.5, я думаю, что лучший подход на самом деле не генератор, а другое решение на основе itertools:
def natatime_25(itr, fillvalue=None, n=2):
x = itertools.chain(iter(itr), (fillvalue,) * (n-1))
return itertools.izip(*(x,)*n)
(поскольку 2.5 не имеет встроенного next
, а также отсутствует izip_longest
).