python - 在 python 中大量使用嵌套循环是否合适？

Question

我有一个项目，我正在尝试做一个复杂的加密方法。大量使用嵌套循环是否正常？还是我错过了什么？

我打算创建一个尝试所有字符串来查找密码的方法。例如，当我输入 ['A','B'] 时，它应该一一创建:

['A', 'B', 'AA', 'AB', 'BA', 'BB', 'AAA', 'AAB', 'ABA', 'ABB', 'BAA', 'BAB', 'BBA', 'BBB', (triple permutations), (quadruple permutations), and it goes on ...]

我的代码:

def rulefinder():
global externalrul1
global externalrul2
rul2 = uniquestring1[0]
rul1 = uniquestring1[0]
for n in range (0,3):
 for m in range (0, len(uniquestring1)):
  for z in range(0, n+1):
    for k in range (0,3):
        for i in range(0, len(uniquestring1)):
            for o in range(0, k+1):
                for y in range (0,  len(uniquestring1)):
                    rul2 = rul2[:-1] + uniquestring1[y]
                    for x in range (0, len(uniquestring1)):
                        rul1= rul1[:-1] + uniquestring1[x]
                        code=""
                        for cha in Genx1:
                            if cha==uniquestring1[0]:
                                code +=codereplacer(rul1)
                            elif cha==uniquestring1[1]:
                                code +=codereplacer(rul2)
                        print(code)
                        print(uniquestring1[0],rul1)
                        print(uniquestring1[1],rul2)
                        print(LastString)
                        if code == LastString:
                            axiom1=uniquestring1[0]
                            axiom2=uniquestring1[1]
                            externalrul1=rul1
                            externalrul2=rul2
                            print('rules are found')
                            print("First RULE:", uniquestring1[0], rul1)
                            print("Second RULE:", uniquestring1[1], rul2)
                            findsubgeneration(code, axiom1, rul1, axiom2, rul2)
                            return
                rul1 = rul1[:o] + uniquestring1[i] + rul1[(o + 1):]
        rul1 += codereplacer(uniquestring1[i])
    rul2 = rul2[:z] + uniquestring1[m] + rul2[(z + 1):]
    rul1 =""
 rul2 += codereplacer(uniquestring1[m])

Answer 1

您正在以一种非常 MATLAB 风格的方式做事(很多循环，并且只在索引上迭代，而不是迭代元素)。 Pythonic 方式更高效(在引擎盖下使用生成器)，并且更简洁:

import itertools
l = ['A','B']
n = 5 # The size of your "expanding permutations"
res = []
for i in range(1,n):
    res.extend(map(''.join, list(itertools.product(l, repeat=i))))
print res

结果:

['A', 'B', 'AA', 'AB', 'BA', 'BB', 'AAA', 'AAB', 'ABA', 'ABB', 'BAA', 'BAB', 'BBA', 'BBB',
'AAAA', 'AAAB', 'AABA', 'AABB', 'ABAA', 'ABAB', 'ABBA', 'ABBB', 'BAAA', 'BAAB', 'BABA', 'BABB', 'BBAA', 'BBAB', 'BBBA', 'BBBB']

警告:只要您使用的是一个小列表，您应该没问题，但随着 l 的增长，结果将呈指数级增长，并且可能会耗尽您的内存。因此，您可以将循环内的结果即时写入磁盘，而不是对 res 列表执行 extend。