python - 查找一个列表的任何元素在另一个列表中出现的索引

Question

获取列表 haystack 和 needles

haystack = ['a', 'b', 'c', 'V', 'd', 'e', 'X', 'f', 'V', 'g', 'h']
needles = ['V', 'W', 'X', 'Y', 'Z']

我需要生成一个索引列表，其中 needles 的任何元素出现在 haystack 中。在这种情况下，这些索引是 3、6 和 8，因此

result = [3, 6, 8]

This question I found非常相似，用

优雅地解决了

result = [haystack.index(i) for i in needles]

不幸的是，在我的例子中，这个解决方案给出了 ValueError: 'W' is not in list。这是因为这里的区别在于 needles 的元素可能在 haystack 中出现多次或根本不出现。

换句话说，haystack 可能不包含针，也可能包含很多。

Answer 1

haystack = ['a', 'b', 'c', 'V', 'd', 'e', 'X', 'f', 'V', 'g', 'h']
needles = ['V', 'W', 'X', 'Y', 'Z']
st = set(needles)
print([i for i, e in enumerate(haystack) if e in st])
[3, 6, 8]

即使您使用了[haystack.index(i) for i in needles if i in haystack]，它也不会工作，因为您有重复的元素。

制作 st = set(needles) 意味着我们有一个线性解决方案，因为集合查找是 0(1)，这对于大输入来说效率会显着提高。