如何创建不包含某些结果的笛卡尔积？

问题描述

为了进行计算，我需要由0和1组成的所有32元素笛卡尔积，其中有十二个1。

我当前正在使用以下方法：

for k,l in enumerate(itertools.product([0,1],repeat=32)):
        if l.count(1)==12:                                        
            # rest code

但是正如您所看到的，对于这么大的笛卡尔乘积来说，它并不是最佳选择。

如何构建我需要的列表，而不必遍历所有元素itertools.product并且没有附加的if条件？有更快的方法吗？

解决方法

这只会生成bits个元素的列表，其中ones是1，其余元素是0：

def binLimit1s( bits,ones,prefix=[] ):
    if bits<ones:
        yield prefix
    elif ones==0:
        yield prefix + [0]*bits
    elif ones==bits:
        yield prefix + [1]*bits
    else:
        for x in binLimit1s( bits-1,prefix+[0] ):
            yield x
        for x in binLimit1s( bits-1,ones-1,prefix+[1] ):
            yield x

在您的情况下，您将使用

binLimit1s( 32,12 )