改写Python中嵌套的for循环

Question

下面的代码有多个循环，我想减少它以优化时间复杂度。

for a in file1:
    if a[0] in [i[1] for i in file2]:
        for b in file2:
            if a[0] == b[1]:
                c.append(int(b[0]))
            continue
    else:
        # do stuff

为了提高效率，我尝试了以下几种方法。不过，我找不到if语句的替代方法。

for a, b in zip(file1, file2):
    if a[0] in [i[1] for i in file2]:
        if a[0] == b[1]:
            c.append(int(b[0]))
        continue
    else:
        # do stuff

此外，这两个操作的输出也不同。第一段代码确实显示了正确的结果。

Answer 1

你的第二个解决方案实际上更慢。zip的思想(或者更确切地说，它应该类似于itertools.product，zip产生N对)产生NxM对，所以您的整个解决方案现在是O(NxMxM)，而第一个解决方案应该是O(Nx2M)。我不确定您的continue语句是做什么的，这似乎毫无意义。

我的建议是预先计算一些值，并使用集合/字典。[i[1] for i in file2]在每个循环中都是一样的，所以把它去掉。

另外，由于您是通过值将b与a对齐，因此让我们创建一个反向查找字典。

# build reverse lookup dictionary
reverse = dict()
for b in file2:
    if not b[1] in reverse:
        reverse[b[1]] = [b]
    else:
        reverse[b[1]].append(b)

# check to see if a[0] matches any b[1], if it does append all matching b[0] to c
for a in file1:
    if a[0] in reverse:
        b_valid = reverse[a[0]]
        for b in b_valid:
            c.append(int(b[0]))
    else:
        # do stuff

这使得它在某种程度上类似于O(N+M) (考虑到糟糕的字典创建时间和查找时间，可能会更糟)。

Answer 2

尝试：

next((x for x in file2 if a[0] == x[1]), None)

这将给你合适的东西，如果它不是None，你应该能够追加它。