python - 在两个 python 列表中查找常见项目的索引

Question

我在 python list_A 和 list_B 中有两个列表，我想找到它们共享的共同项目。我这样做的代码如下:

both = []
for i in list_A:
    for j in list_B:
        if i == j:
            both.append(i)

最后的common 列表包含了常见的项目。但是，我还想在最初的两个列表中返回这些元素的索引。我该怎么做？

Answer 1

如果有更好的方法可用，建议在 python 中尽可能避免使用 for 循环。如下使用python set可以高效的找到两个列表中的共同元素

both = set(list_A).intersection(list_B)

然后你可以使用内置的 index 方法找到索引

indices_A = [list_A.index(x) for x in both]
indices_B = [list_B.index(x) for x in both]