生成具有固定位置的二进制组合Python

问题描述

我想生成大小为 n=16 的 0/1 的所有可能组合的元组列表，但其中某些位置是固定的。所以有些位置固定为 0 或 1 (x,x,1,x)。我知道我们可以使用 itertools 来生成没有任何固定位置的所有可能组合：

import itertools

permutations = list(itertools.product([0,1],repeat=16))

我想用 repeat=13 生成所有组合，然后插入固定位置，但这对我来说似乎太麻烦了。是否有更简单快捷的方法来实现这一目标？

解决方法

只是基于 itertools.product 的一点点变化：

def custom_product(f,repeat=0):
    return itertools.product(*[f.get(i,[0,1]) for i in range(repeat)])

fixed = {4: [1],8: [0],9: [0]}
print(*custom_product(fixed,16),sep="\n")

并且结果的格式与您预期的相同。

,

我喜欢递归：

n_bits = 8
fixed = {0:"0",3:"1"}

def permut_fixed(n_bits,fixed):
    def aux(base,bits_left):
        if bits_left<=0:
            return [base]
        if len(base) in fixed:
            return aux(base+fixed[len(base)],bits_left-1)
        else:
            return [*aux(base+"0",bits_left-1),*aux(base+"1",bits_left-1)]
    return aux("",n_bits)

print(permut_fixed(n_bits,fixed))
# ['00010000','00010001',...,'01111011','01111100','01111101','01111110','01111111']
print(len(permut_fixed(n_bits,fixed)))
# 64 which is 2**6 and we have 6 slots to fill