在Python位置的列表之间匹配元素

Question

我有两张单子，都很长。列表A包含一个整数列表，其中一些整数在列表B中重复出现。我可以通过以下方法找到出现在这两个列表中的元素：

idx = set(list_A).intersection(list_B)

这将返回列表a和列表B中出现的所有元素的集合

但是，我想找到一种方法来查找两个列表之间的匹配，并保留有关元素在两个列表中的位置的信息。这样的函数可能看起来像：

def match_lists(list_A,list_B):
.
.
.
return match_A,match_B

其中match\u A将包含列表\u A中元素的位置，这些元素在列表\u B中的某个位置有匹配项，反之亦然。 我可以看到如何使用for循环来构造这样的列表，但是对于长列表来说，这样做的速度会非常慢。你知道吗

关于重复项：列表\u B中没有重复项，如果列表\u a中有重复项，则返回所有匹配的位置作为列表，因此匹配\u a将是列表的列表。你知道吗

Answer 1

这个怎么样：

def match_lists(list_A, list_B):
    idx = set(list_A).intersection(list_B)
    A_indexes = []
    for i, element in enumerate(list_A):
        if element in idx:
            A_indexes.append(i)
    B_indexes = []
    for i, element in enumerate(list_B):
        if element in idx:
            B_indexes.append(i)
    return A_indexes, B_indexes