Python的电源设置操作



我正在操纵一个集合,因此,如果您具有n个独特元素的集合(又称:列表(,则您有2^n子集。我在这里展示:

def powerset(s):
    x = len(s)
    masks = [1 << i for i in range(x)]
    for i in range(1 << x):
        yield [ss for mask, ss in zip(masks, s) if i & mask]
l = list(powerset(["A", "B"]))
print(l) 

给出:

[[], ['A'], ['B'], ['A', 'B']]

现在如何获取上述列表消除空列表,并合并最后一个元素,以使其变为:

['A', 'B', 'AB']

我想重复此过程5次,取出最终输出并写下其冠军,消除空列表并合并它们属于同一sublist的元素。

要摆脱空套,只需使用1而不是0启动循环,然后''.join

def powerset(s):
    x = len(s)
    masks = [1 << i for i in range(x)]
    for i in range(1, 1 << x):
        yield ''.join(ss for mask, ss in zip(masks, s) if i & mask)

如果要重复此操作,即获取原始列表的PowerSet的功能,只需将结果馈回循环中的功能:

lst = ["A", "B"]
for _ in range(5):
    lst = list(powerset(lst))
    print(lst)

话虽如此,这样的过滤和加入作为后处理步骤可能更有意义,就像在 @l3viathan的答案中一样,因为真正的powerset功能不应忽略或修改结果。

data = [[], ['A'], ['B'], ['A', 'B']]
list(filter(None,map(lambda x:''.join(x) if x else None, data)))
>>>['A', 'B', 'AB']

首先,滤除虚假(空(元素,然后加入其余元素的元素:

>>> l = [[], ['A'], ['B'], ['A', 'B']]
>>> list(map(''.join, filter(bool, l)))
['A', 'B', 'AB']

等效的清单透明方式:

>>> l = [[], ['A'], ['B'], ['A', 'B']]
>>> [''.join(e) for e in l if e]
['A', 'B', 'AB']

要做五次,这样做五次:

start = ["A", "B"]
for _ in range(5):
    start = [''.join(e) for e in powerset(start) if e]

也许您需要诸如flatmap之类的东西?

from itertools import chain, imap
def flatmap(f, items):
        return chain.from_iterable(imap(f, items))
>>> list(flatmap(lambda x: x, [[], ['A'], ['B'], ['A', 'B']]))
['A', 'B', 'A', 'B']

这是您要寻找的:

def powerset(s):
    x = len(s)
    masks = [1 << i for i in range(x)]
    for i in range(1 << x):
        item = ''.join([ss for mask, ss in zip(masks, s) if i & mask])
        if item:
            yield item
l = list(powerset(["A", "B", "C"]))
print(l) 
#['A', 'B', 'AB', 'C', 'AC', 'BC', 'ABC']

我引入了C,因为它显然与AB一起使用。

最新更新