如何比較兩個具有不同長度的列表並刪除一些元素？

Question

我有兩個列表，第一個包含一些在第二個列表中不存在的元素，這些元素具有更多元素。 我需要刪除第二個列表中不存在的元素。

在下面的示例中，我只需要從第一個列表中刪除“ 0”即可。 所以：

Input:
list1 = [0,1,2,3]
list2 = [1,2,3,4,5,6,7]

Output:
list1 = [1,2,3]
list2 = [1,2,3,4,5,6,7]

我嘗試了這個簡單的代碼，但它返回了一個空列表

list1=[0,1,2,3]
list2=[1,2,3,4,5,6,7]

for element in list1:
    for element1 in list2:
        if element != element1:
            for element in list1:
                list1.remove(element)
print(list1)

有什么幫助嗎？

Answer 1

您可以使用集來做到這一點！

set(list1).intersection(list2) # [1,2,3]

您也可以使用列表推導，但是理論上它比較慢。

list1 = [0,1,2,3]
list2 = [1,2,3,4,5,6,7]

list1 = [x for x in list1 if x in list2] # [1,2,3]