如何根据 Python 中的特定条件在更大的列表中分解列表？

Question

我有一个列表my_list = [[1, 2, 3, 4], [8, 9], [89, 90]] 。 我想通过扫描较大列表中的每个列表来分解它。 如果满足条件（例如，如果找到3 ），我想将该内部列表分成两个列表。

期望的最终结果应该是[[1, 2], [3, 4], [8, 9], [89, 90]] 。

但我的代码如下：

def break_list_by_specific_number(list_of_lists, break_num):
    list_holder = []
    for each_list in list_of_lists:
        i = (list(g) for _, g in groupby(each_list, key=break_num.__ne__))
        list_holder += ([a + b for a, b in zip_longest(i, i, fillvalue=[])])
    return list_holder

print(break_list_by_specific_number(my_list, 3))

它错误地将其分解为[[1, 2, 3], [4], [8, 9], [89, 90]] 。

Answer 1

你可以这样写得更快、更简单：

def break_list(array, break_num):
    out = []
    for inner in array:
        if break_num in inner:
            index = inner.index(break_num)
            first = inner[:index]
            second = inner[index:]
            out.append(first)
            out.append(second)
        else:
            out.append(inner)
    return out

这假设inner最多有 1 个break_num实例，并且该array是一个二维列表。

你调用它的方式与你最初调用它的方式相同，它工作得很好。

>>> arr = [[1, 2, 3, 4], [8, 9], [89, 90]]
>>> break_list(arr, 3)
[[1, 2], [3, 4], [8, 9], [89, 90]]