获取长度为 3 [0,0,0] 到 [1,1,1] 的 [1,0] 的所有可能组合

Question

    from itertools import combinations
   
     
def n_length_combo(arr, n):
     
    # using set to deal
    # with duplicates 
    return list(combinations(arr, n))
   
# Driver Function
if __name__ == "__main__":
    arr = '01'
    n = 3
    print (n_length_combo([x for x in arr], n) )

预期输出

想要 0 和 1 的 3 个组合。尝试了上面的示例，但它不起作用

Answer 1

您正在寻找笛卡尔积，而不是 的组合或排列[0, 1]。为此，您可以使用itertools.product.

from itertools import product

items = [0, 1]

for item in product(items, repeat=3):
    print(item)

这会产生您正在寻找的输出（尽管顺序略有不同）：

(0, 0, 0)
(0, 0, 1)
(0, 1, 0)
(0, 1, 1)
(1, 0, 0)
(1, 0, 1)
(1, 1, 0)
(1, 1, 1)