list_1 = ["a", "b", "c", "d", "e"]
list_2 = ["a", "c", "d"]
我想要获取在
list_1
中但不在list_2
中的元素索引。我期望结果如下:[1, 4]
对于我的情况,list_2
是 list_1
的一个子集并且所有元素都是唯一的。是否有更好的方法可以在不使用显式循环的情况下完成此操作?
你可以使用条件列表推导式:
>>> [i for i, item in enumerate(list_1) if item not in list_2]
[1, 4]
这个解决方案的时间复杂度为O(n*m)
。对于更大的列表,将list_2
转换为set
是有意义的,因为在set
中搜索速度更快。以下解决方案的时间复杂度为O(n)
:
>>> set_2 = set(list_2)
>>> [i for i, item in enumerate(list_1) if item not in set_2]
[1, 4]