(Python 3) 使用唯一元素作为分隔符拆分列表

Question

我有一个列表 l=[2,3,4,5,6,99,7,8,9,10,11,12,100,13,14,15,16,17,18,101]，我想将 l 拆分为子列表 [2,3,4,5,6], [7,8,9,10,11,12],[13,14,15,16,17,18]，这意味着我使用属于 separators = [99,100,101] 的“分隔符”数字 99,100,101 作为标志来指示我应该去哪里到下一个列表。

特别是，这些子列表可能具有不同数量的元素，但大小仅相差 1（5 或 6）。注意：分隔符可能超过 3 个。

在 Python 3 中有没有一种有效的方法来做到这一点？我想先提取分隔元素的索引，然后相应地对列表进行切片，但这似乎太复杂且计算量太大了..

一些见解会很棒！

添加（来自@Netwave 的建议）：我的尝试（显然行不通）：

g = []
for i in l:
    if i in separators:
        g += [l[:l.index(i)]]

输出：

>>> g
[[2, 3, 4, 5, 6], [2, 3, 4, 5, 6, 99, 7, 8, 9, 10, 11, 12], [2, 3, 4, 5, 6, 99, 7, 8, 9, 10, 11, 12, 100, 13, 14, 15, 16, 17, 18]]

Answer 1

使用groupby:

from itertools import groupby

separators = [99, 100, 101]
l = [2, 3, 4, 5, 6, 99, 7, 8, 9, 10, 11, 12, 100, 13, 14, 15, 16, 17, 18, 101]

splits = [list(values) for key, values in groupby(l, key=lambda x: x not in separators) if key]

print(splits)

输出

[[2, 3, 4, 5, 6], [7, 8, 9, 10, 11, 12], [13, 14, 15, 16, 17, 18]]

Answer 2

我希望您正在寻找类似于以下代码的内容。您需要遍历您的列表并继续检查该元素是否存在于分隔符列表中。这可以通过其他方式完成，正如您所说，通过遍历分隔符列表并找到主列表中元素的索引。为了简单起见，我采用了前一种方法。 （注意endindex变量的使用）：

l=[2,3,4,5,6,99,7,8,9,10,11,12,100,13,14,15,16,17,18,101]
seperator = [99,100,101]

list_of_list = []
endindex = 0
for i in range(0,len(l),1):
    if l[i] in seperator:
        list_of_list.append(l[endindex:i])
        endindex = i + 1

print(list_of_list)

输出：

[[2, 3, 4, 5, 6], [7, 8, 9, 10, 11, 12], [13, 14, 15, 16, 17, 18]]

Answer 3

更容易在函数中使用：

import itertools
def split(l,l2):
    return [list(v) for k,v in itertools.groupby(l,lambda x: x in l2) if not k]
l = [2, 3, 4, 5, 6, 99, 7, 8, 9, 10, 11, 12, 100, 13, 14, 15, 16, 17, 18, 101]
print(split(l,[99, 100, 101]))

输出：

[[2, 3, 4, 5, 6], [7, 8, 9, 10, 11, 12], [13, 14, 15, 16, 17, 18]]

实现了Split a list into nested lists on a value的复制