改写 Python 中的嵌套 for 循环

如何解决改写 Python 中的嵌套 for 循环

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

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

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

解决方法

您的第二个解决方案实际上更慢。 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) 的某个位置（鉴于糟糕的字典创建时间和查找时间，可能会更糟）。

,

试试：

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

这会给你合适的，如果不是 None，你应该可以追加。