从列表中删除嵌套(2个元素的列表)的简单方法?

时间:2015-08-23 11:50:00

标签: python

我有input_list = [[1,2],[1,3],[1,4],[2,8],[2,10]]

然后我有elimination_list = [[1,2],[2,8]]

然后我创建一个result_list,其中包含输入列表的元组,但在elimination_list中找到的元组除外

因此result_list = [[1,3],[1,4],[2,10]]

我可以通过下面的多个for循环来实现这一点,但是有什么方法可以用较少的代码行来做到这一点吗?

for x,y in input_list:
    is_eliminate = 0
    for a,b in elimination_list:
        if x == a and y == b:
            is_eliminate = 1

    if is_eliminate == 0:
        result_list.append([x,y])

3 个答案:

答案 0 :(得分:5)

您可以使用包含测试(in)和列表理解:

result_list = [t for t in input_list if t not in elimination_list]

t not in elimination_list否定收容测试会遍历所有elimination_list,以便您查看该元素是否存在。

如果您可以将列表转换为实际具有元组,则可以使用设置成员资格测试的更快选项:

input_list = [(1, 2), (1, 3), (1, 4), (2, 8), (2, 10)]
elimination_set = {(1, 2), (2, 8)}
result_list = [t for t in input_list if t not in elimination_set]

此处eliminitaion_set是一个包含元组(不可变序列)的集合,t not in elimination_set使用散列算法来确定任何给定元组是否是平均O(1)(常量)时间的成员。 / p>

你不 将input_list转换为元组,但是;只有在测试时才能进行转换:

input_list = [[1, 2], [1, 3], [1, 4], [2, 8], [2, 10]]
elimination_set = {(1, 2), (2, 8)}
result_list = [t for t in input_list if tuple(t) not in elimination_set]

答案 1 :(得分:3)

您可以将列表理解与条件一起使用:

result_list = [t for t in input_list if t not in elimination_list]

对于较大的列表,您可以使用set来更快地消除:

elimination_set = set((tuple(x) for x in elimination_list))
result_list = [t for t in input_list if tuple(t) not in elimination_set]

功能风格的两个解决方案:

result_list = filter(lambda t: t not in elimination_list, input_list)

或:

elimination_set = set(map(lambda x: tuple(x), elimination_list))
result_list = filter(lambda t: t not in elimination_set, input_list)

答案 2 :(得分:2)

只需使用filterin

result = filter(lambda x: x not in elimination_list, input_list)