python - 在 2 个列表中搜索相同的值，但如果重复则只搜索一次

Question

lst1 = [0.0, 2.0, 2.0, 3.0, 4.0, 6.0, 6.0, 7.0, 8.0, 8.0, 10.0, 10.0, 10.0, 13.0]
lst2 = [4.,  8.]
[i for i, j in enumerate(lst1) if j in lst2]

我会得到返回:

[4, 8, 9]

我想要实现的是我想要的返回:

[4, 8]

因为 8.0 在 lst1 中重复，我只想要一次。实现这一目标的最佳方式是什么？

Answer 1

我能想到的最好的方法是，维护一组已经看到的数字，如果该数字不在已经看到的 seen 集合中，则将 index 添加到result 和 seen 集合的编号。

result, seen = [], set()
for index, item in enumerate(lst1):
    if item not in seen and item in lst2:
        result.append(index)
        seen.add(item)
print(result)
# [4, 8]

注意 1: 我们使用 set 来维护 seen 项目，因为查找会非常快(恒定时间查找)

注意 2:如果您的 lst2 也将是一个长列表，那么最好将其转换为一个集合，以便 lst2 中的项目 查找也将在恒定时间内发生。所以实际的代码变成了，

result, seen, set2 = [], set(), set(lst2)
for index, item in enumerate(lst1):
    if item not in seen and item in set2:
        result.append(index)
        seen.add(item)
print(result)
# [4, 8]

如果 lst2 会很长，这将提供比第一个版本更好的性能。

受 DSM 解决方案启发的替代解决方案是，构建一个反向索引字典，如下所示

>>> reverse_indexes = {lst1[i]: i for index in xrange(len(lst1) - 1, -1, -1)}

然后我们可以像这样简单地进行查找

>>> [reverse_indexes[item] for item in lst2 if item in reverse_indexes]
[4, 8]