python-3.x - 将列表拆分为特定值的多个列表

Question

我有一个内部由 0 分隔的元素列表。 格式是这样的:

[0, 2, 2, 0, 1, 5, 5, 3, 1, 3, 0, 7, 4, 6, 7, 4, 6]

我想使用零作为分隔符并将其拆分为子列表。在那个例子中:

[[], [2, 2], [1, 5, 5, 3, 1, 3], [7, 4, 6, 7, 4, 6]]

因此，零将消失。 前导零或结尾零应在开头或结尾形成一个空列表。

直到知道，我通过将列表连接到一个字符串，然后拆分它并再次制作一个列表来以一种丑陋的方式做到这一点。

是否有一种奇特的 pythonic 方法可以更轻松地做到这一点？

Answer 1

您可以使用 groupby 方法:

from itertools import groupby
k = [0, 2, 2, 0, 1, 5, 5, 3, 1, 3, 0, 7, 4, 6, 7, 4, 6]

gr = groupby(k, lambda a: a==0)

l = [[] if a else [*b] for a,b in gr]

print(l)

输出:

[[], [2, 2], [], [1, 5, 5, 3, 1, 3], [], [7, 4, 6, 7, 4, 6]]

您可以通过以下方式修复内部 []:

l = [ a for idx,a in enumerate(l) if idx in (0,len(l)) or a]

得到

[[], [2, 2], [1, 5, 5, 3, 1, 3], [7, 4, 6, 7, 4, 6]]

---

您可以将列表 comp 替换为

l = []
for a,b in gr:
    if a: 
        l.append([])
    else: 
        l.append([*b])