python - 删除另一个列表中存在的一个列表中的元素

Question

我有list1和list2。list2是一组必须从 中删除的单词，list1例如：

list1=['paste', 'text', 'text', 'here', 'here', 'here', 'my', 'i', 'i', 'me', 'me']

list2=["i","me"]

期望的输出：

list3=['paste', 'text', 'text', 'here', 'here', 'here', 'my']

我尝试过使用'for'的不同版本，但到目前为止没有结果。

任何想法，将不胜感激！

Answer 1

使用列表理解：

>>> list1 = ['paste', 'text', 'text', 'here', 'here', 'here', 'my', 'i', 'i', 'me', 'me']
>>> list2 = ["i","me"]
>>> list3 = [item for item in list1 if item not in list2]
>>> list3
['paste', 'text', 'text', 'here', 'here', 'here', 'my']

注意：列表中的查找是O(n)，请考虑从中创建一个集合list2- 集合中的查找是O(1).

Answer 2

如何利用集合算术？

diff = set(list1) - set(list2)
result = [o for o in list1 if o in diff]

甚至更好（更有效）：

set2 = set(list2)
result = [o for o in list1 if o not in set2]