Python 3：如何根据条件拆分列表？

Question

我已经找到了很多关于如何将列表分成大小相等的块的答案，但我有一个不同的问题

我有如下格式的数据

> header1
line1
line2
...
> header2
line4
line5
...

我想把这些行分组在各自的标题下

获取标题很容易headers = [x for x in lines if x.startswith('>')]

但这个技巧对后续的行不起作用，因为无法知道每个标题下有哪些行

理想情况下，我想要一个格式为[[line1, line2], [line4, line5]...]的列表

我有一个使用while循环的工作解决方案，但它看起来很难看。我如何使用列表理解或现有库来实现这一点

Answer 1

将itertools.groupby与自定义键函数一起使用，该函数在每次看到新头时都会更改。在这个函数中，我们增加ctr

from itertools import groupby

lis = ['>a', 'b', 'c', '>d', 'e', '>f', '>g']

def group_by_header(lis: list):
    def header_counter(x: str):
        if x.startswith('>'):
            header_counter.ctr += 1
        return header_counter.ctr
    header_counter.ctr = 0

    return groupby(lis, key=header_counter)

print([list(l) for k, l in group_by_header(lis)])
# [['>a', 'b', 'c'], ['>d', 'e'], ['>f'], ['>g']]