从python中的两个列表中删除相同的元素

Question

我正在编写一个函数来删除两个列表中相同的元素。

 def del_same_elements(num1,num2):
        for i in range(0,len(num1)):
            for j in range(0,len(num2)):

                if num1[i] == num2[j]:
                    same = num1[i]

        num1.remove(same)
        num2.remove(same)

        return num1,num2

调用print (del_same_elements([3,11],[2,2,3]))按预期返回[11],[2,2] ，但在尝试print (del_same_elements([3],[2,2]))我local variable 'same' referenced before assignment收到错误local variable 'same' referenced before assignment 。 我如何处理没有相同值的情况？

Answer 1

使用集合intersection来检测公共元素：

def rm_same(num1, num2):
    same = set(num1).intersection(num2)
    for s in same:
        num1.remove(s)
        num2.remove(s)
    return num1, num2

Answer 2

使用一些集合论并计算列表的交集，然后删除项目交集

def del_same_elements(num1, num2):
    same = list(set(num1) & set(num2))

    for i in same:
        num1.remove(i)
        num2.remove(i)

    return num1, num2

if __name__ == "__main__":
    num1, num2 = del_same_elements([1,2,3,4,5], [1,3,5,6])

print num1
print num2

结果

[2, 4]
[6]

这在没有交叉的地方也很有效，因为什么都不会被删除

Answer 3

或者，您可以编写这样的程序

def del_same_elements(num1,num2):
    for num in num1 + num2:
        if num in num1 and num in num2:
            num1.remove(num)
            num2.remove(num)

    return num1, num2

print (del_same_elements([3,11],[2,2, 3]))

这可能是一个更好的解决方案。

Answer 4

或者，您可以改用集合：

same = set(num1) & set(num2)

while same:
    val = same.pop()
    num1.remove(val)
    num2.remove(val)

这至少避免了那些嵌套循环。