Python 减少条件表达式

Question

我有 9 个变量 a、b、c、d、e、f、g、h、i，我将它们循环到 9 for 循环中，从 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 循环的范围不同怎么办？

提前致谢！

Answer 1

您可以使用 itertools 模块简单地执行此操作：

from itertools import permutations

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

Answer 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

what if i want sequence of abcdefghi such taht a,b,c,e,g is value from 0 to 9, and d,f,h,i in the range of 1 to 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 (for intro see 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

Python 减少条件表达式

Python Reduce conditional expression

python

conditional-statements