python - 计算方程的可能组合

Question

我有一个方程式模板 template = (a op b) op c 其中可用的操作 op 如下: op = [' +', '-', '/', '*']。

我想做的是使用操作 op 和参数 a、b 和c。

例如

(a + b) + c
(a - b) - c
(a * b) * c
(a / b) / c
(a + c) + b
(a - c) + b
(b * a) - c
...
...
...

我查看了 itertools.combinations 但是我不太确定我是否(或如何)可以在这个特定的上下文中使用这个函数。

Answer 1

这就是你想要的:

import itertools

symbol = ['a', 'b', 'c']
op = ['+', '-', '/', '*']

for symbols in itertools.permutations(symbol):
    for ops in itertools.product(op, repeat=2):
        print "(%s %s %s) %s %s" % (
            symbols[0], ops[0], symbols[1], ops[1], symbols[2])