從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)

這至少避免了那些嵌套循環。