Python减少条件表达式

时间:2015-03-08 22:33:28

标签: python conditional-statements

我有9个变量a,b,c,d,e,f,g,h,i和I将它们循环到9里面,循环从0到9.但是范围可能会有所不同。

我希望他们的所有序列都是abcdefghi,这样就没有重复的数字了。

现在我有了这个,下面:

for a in range(0, 9): 
    for b in range(0,9): #it doesn't have to start from 0
    ....
        for i in range(0, 9):
             if a != b and a != c ... a != i
                b != c and b != d ... b != i
                c != d and c != e ... c != i
                ... h != i:

                print (a,b,c,d,e,f,g,h,i)

有9个! = 362880,

但是我怎样才能减少条件表达式呢?如果for循环的范围不同会怎么样?

提前致谢!

2 个答案:

答案 0 :(得分:2)

您可以使用itertools模块执行此操作:

from itertools import permutations

for arrangement in permutations('abcdefghi', 9):
    print ''.join(arrangement)

答案 1 :(得分:2)

from itertools import permutations

for perm in permutations(range(1, 10), 9):
    print(" ".join(str(i) for i in perm))

给出了

1 2 3 4 5 6 7 8 9
1 2 3 4 5 6 7 9 8
1 2 3 4 5 6 8 7 9
1 2 3 4 5 6 8 9 7
1 2 3 4 5 6 9 7 8
1 2 3 4 5 6 9 8 7

# ... etc - 9! = 362880 permutations

  

如果我想要abcdefghi这样的序列a,b,c,e,g是从0到9的值,并且d,f,h,i在1到5的范围内

这有点复杂,但仍然可以实现。首先在d..i中选择值更容易:

from itertools import permutations

for d,f,h,i,unused in permutations([1,2,3,4,5], 5):
    for a,b,c,e,g in permutations([unused,6,7,8,9], 5):
        print(a,b,c,d,e,f,g,h,i)

给出了

5 6 7 1 8 2 9 3 4
5 6 7 1 9 2 8 3 4
5 6 8 1 7 2 9 3 4
5 6 8 1 9 2 7 3 4
5 6 9 1 7 2 8 3 4
5 6 9 1 8 2 7 3 4
5 7 6 1 8 2 9 3 4
5 7 6 1 9 2 8 3 4
5 7 8 1 6 2 9 3 4
5 7 8 1 9 2 6 3 4

# ... etc - 5! * 5! = 14400 permutations

对于一般情况(即数独),您需要一个更通用的解决方案 - 约束求解器,如python-constraint(对于介绍,请参阅the python-constraint home page)。

然后你的解决方案开始看起来像

from constraint import Problem, AllDifferentConstraint

p = Problem()
p.addVariables("abceg", list(range(1,10)))
p.addVariables("dfhi",  list(range(1, 6)))
p.addConstraint(AllDifferentConstraint())

for sol in p.getSolutionIter():
    print("{a} {b} {c} {d} {e} {f} {g} {h} {i}".format(**sol))

给出了

9 8 7 4 6 3 5 2 1
9 8 7 4 5 3 6 2 1
9 8 6 4 7 3 5 2 1
9 8 6 4 5 3 7 2 1
9 8 5 4 6 3 7 2 1
9 8 5 4 7 3 6 2 1
9 7 8 4 5 3 6 2 1
9 7 8 4 6 3 5 2 1
9 7 6 4 8 3 5 2 1
9 7 6 4 5 3 8 2 1
9 7 5 4 6 3 8 2 1

# ... etc - 14400 solutions