假设我有一个字符串列表：
l = [‘A’,’B’,’C’,’D’]

我知道要生成所有带有替换的组合,请选择n,我将使用itertools.combinations库方法来获取它们.

例如,
list(combinations(l,2))会给我

[[‘A’,’B’],[‘A’,’C’],[‘A’,’D’],[‘B’,’C’],[‘B’,’D’] ,[‘光盘’]]

但是,我想用括号代替括号：

[‘(A,B)’,'(A,C)’,'(A,D)’,'(B,C)’,'(B,D)’,'(C,D)’]

现在,假设我要扩展它,并为这些AND和OR添加两个操作：

这样我就可以得到[‘(A’,’AND’,’B)’,'(A’,’OR’,’B)’等]

进一步扩展,以在n = 3的情况下获得嵌套的括号：

['((A','AND','B)', 'AND', 'C)', '((A','AND','B)', 'OR', 'C)', '((A','OR','B)', 'OR', 'C)', '((A','OR','B)', 'AND', 'C)', etc.]

理想的形式是：

['((A AND B) AND C)', '((A AND B) OR C)', '((A OR B) OR C)', '((A OR B) AND C)', etc.]

因此,总而言之,我一次选择一个列表n个元素的组合,并在运算符[‘AND’,’OR’]上进行排列,并从左侧添加嵌套.

我已经用JavaScript做过类似的事情,但是这样做很容易,因为用户可以构建实际的句子.它不是由一组排列和组合创建的.

解决方法:

您可以使用itertools.combinations从给定列表中选择操作数,使用itertools.product生成操作符组合,再次使用itertools.product生成操作数和操作符的所有混合,并使用for循环将嵌套列表根据选择的操作数和运算符以构建所需的输出：

from itertools import combinations, product
def expressions(l, n):
    for (operations, *operands), operators in product(
            combinations(l, n), product(('AND', 'OR'), repeat=n - 1)):
        for operation in zip(operators, operands):
            operations = [operations, *operation]
        yield operations

因此list(expressions([[‘A’,’B’,’C’,’D’],3))返回：

[[['A', 'AND', 'B'], 'AND', 'C'],
 [['A', 'AND', 'B'], 'OR', 'C'],
 [['A', 'OR', 'B'], 'AND', 'C'],
 [['A', 'OR', 'B'], 'OR', 'C'],
 [['A', 'AND', 'B'], 'AND', 'D'],
 [['A', 'AND', 'B'], 'OR', 'D'],
 [['A', 'OR', 'B'], 'AND', 'D'],
 [['A', 'OR', 'B'], 'OR', 'D'],
 [['A', 'AND', 'C'], 'AND', 'D'],
 [['A', 'AND', 'C'], 'OR', 'D'],
 [['A', 'OR', 'C'], 'AND', 'D'],
 [['A', 'OR', 'C'], 'OR', 'D'],
 [['B', 'AND', 'C'], 'AND', 'D'],
 [['B', 'AND', 'C'], 'OR', 'D'],
 [['B', 'OR', 'C'], 'AND', 'D'],
 [['B', 'OR', 'C'], 'OR', 'D']]