Python二进制与列表值比较

Question

我想弄清楚如何在 Python 中做到这一点：

打印出任何给定集合的所有子集。

例如：[1,2]

所以答案是手动的：

[]
[2]
[1]
[1,2]

我意识到解决方案的总数将是2(exp)n，其中n 是列表中元素的数量。

例如，如果列表是[1,3,4]，那么子集的总数就是2(exp)3 = 8。

我也意识到，如果我得到上面列表的二进制位表示，就会出现以下内容：

例如：[1,2]

00 : []
01 : [2]
10 : [1]
11 : [1,2]

包含1 的位的每个位置，即子集在将其索引到原始集[1,2] 时的位置。例如二进制 01 = 在原始集合 [1,2] 的位置 1 处获取索引，这将是 [2]。

二进制 11，表示从原始集合 [1,2] 中获取索引位置 0 和 1，其答案为 [1,2]，等等。

我该如何编码，我的代码太乱了，有没有简单的方法来映射这个？

Answer 1

您可以使用循环遍历所有二进制组合和嵌套循环测试计数器的每一位来决定是否将相应索引的列表元素添加到输出列表：

def combinations(l):
    for i in range(1 << len(l)):
        c = []
        for b in range(len(l)):
            if i & (1 << (len(l) - b - 1)):
                c.append(l[b])
        yield c

或使用列表理解：

def combinations(l):
    return [[l[b] for b in range(len(l)) if i & (1 << (len(l) - b - 1))] for i in range(1 << len(l))]

这样：

list(combinations([1, 2])))

会返回：

[[], [2], [1], [1, 2]]

和：

list(combinations([1, 3, 4]))

会返回：

[[], [4], [3], [3, 4], [1], [1, 4], [1, 3], [1, 3, 4]]