Question

我有两个列表。我想获取匹配的索引。我提出了以下示例作为示例：

a = [1,2,6,5,3,4]
b = [1,3,4,2,5,6]

for i in range(len(a)):
    for j in range(len(b)):
        if b[j] != a[i]:
            next
        else:
            print(i, j)

似乎很简单，但是有点长。谁能帮我提高代码效率？

输出：

Answer 1

a = [1,2,6,5,3,4]
b = [1,3,4,2,5,6,1,2,3,4,5,6]

b_dict = {}
for i, n in enumerate(b):
    if n not in b_dict:
        b_dict[n] = []
    b_dict[n].append(i)

for i in range(len(a)):
    if a[i] in b_dict:
        for index in b_dict[a[i]]:
            print(i, index)

O（a + b）解决方案，因为字典中的“ in”为O（1）

更新版本允许在b中匹配重复的数字

Answer 2

您可以像这样使用列表理解：

>>> a = [1,2,6,5,3,4]
>>> b = [1,3,4,2,5,6]
>>>
>>> [[i,j] for i,k in enumerate(a) for j,v in enumerate(b) if k==v]
[[0, 0], [1, 3], [2, 5], [3, 4], [4, 1], [5, 2]]

Answer 3

提示

如果您有一个使用索引及其各自值的循环，请使用enumerate而不是range(len())和索引。
在这种情况下，您无需为不匹配条件（if b[j] != a[i]）操心。

>>> for i, v0 in enumerate(a):
...     for j, v1 in enumerate(b):
...         if v0 == v1:
...             print(i, j)
... 
0 0
1 3
2 5
3 4
4 1
5 2

作为列表理解：

>>> [(i, j) for i, v0 in enumerate(a) for j, v1 in enumerate(b) if v0 == v1]
[(0, 0), (1, 3), (2, 5), (3, 4), (4, 1), (5, 2)]

（我输出的元组是因为在这种情况下它们似乎更有意义。）

如何使此匹配列表代码更有效？

3 个答案: