python - 如何根据分隔符将列表拆分为子列表，类似于 str.split()？

Question

给定如下列表:

[a, SEP, b, c, SEP, SEP, d]

我如何将其拆分为子列表列表:

[[a], [b, c], [], [d]]

实际上我需要一个等价于 str.split() 的列表。我可以拼凑一些东西，但我似乎无法想出任何整洁和/或 pythonic 的东西。

我从迭代器获得输入，因此生成器也可以接受。

更多例子:

[a, SEP, SEP, SEP] -> [[a], [], [], []]

[a, b, c] -> [[a, b, c]]

[SEP] -> [[], []]

Answer 1

一个简单的生成器将适用于您问题中的所有情况:

def split(sequence, sep):
    chunk = []
    for val in sequence:
        if val == sep:
            yield chunk
            chunk = []
        else:
            chunk.append(val)
    yield chunk