组合式的写法

Question

I want to find all possible combination of signs of a given equation, eg Input (q,p1). Output: -p1+q, p1+q, p1-q, -p1 -q.

Does anybody know how to do this in python?

Answer 1

you can use the product function (from itertools) to create the sign combinations and concatenate them to the terms using zip and join:

from itertools import product
terms = ["p","q1"]
opers = product(*["+-"]*len(terms))
formulas = ["".join(o+t for o,t in zip(op,terms)).strip("+") for op in opers]

print(formulas)
# ['p+q1', 'p-q1', '-p+q1', '-p-q1']