python - 为什么 takewhile() 会跳过第一行？

Question

我有这样一个文件:

1
2
3
TAB
1
2
3
TAB

我想以 block 的形式读取 TAB 之间的行。

import itertools

def block_generator(file):
    with open(file) as lines:
        for line in lines:
            block = list(itertools.takewhile(lambda x: x.rstrip('\n') != '\t',
                                             lines))
            yield block

我想这样使用它:

blocks = block_generator(myfile)
for block in blocks:
    do_something(block)

我得到的 block 都是从第二行开始的，比如 [2,3] [2,3]，为什么？

Answer 1

这是使用 groupby 的另一种方法

from itertools import groupby
def block_generator(filename):
    with open(filename) as lines:
        for pred,block in groupby(lines, "\t\n".__ne__):
            if pred:
                yield block