python 如何写多重循环

Question

如下图流程图，应该如何写循环完成。求指点

Answer 1

流程图已经很详细了，照着把它翻译成Python代码就OK了

A = [ [ 1, 4 ], [ 2, 5 ], [ 3, 6 ] ]
a = [ 1, 2, 3, 4, 5, 6 ]
B = [ [ 4, 1 ], [ 1, 5 ], [ 2, 5 ], [ 1, 7 ], [ 7, 8 ] ]

list1, list2, list3, list4 = [ ], [ ], [ ], [ ]
for i in B:
    if i in A:
        list1.append(i)
    elif i[ ::-1 ] in A:
        list1.append(i)
    elif i[ 0 ] in a and i[ 1 ] in a:
        list2.extend(i)
    elif i[ 0 ] in a and i[ 1 ] not in a:
        list3.extend(i)
    elif i[ 0 ] not in a and i[ 1 ] not in a:
        list4.extend(i)

print(list1)
print(list2)
print(list3)
print(list4)

Answer 2

道理都差不多，你看下逻辑是否有最优化的就行，无非就是 if elif ...else