我正在操纵一个集合,因此,如果您具有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
,因为它显然与A
和B
一起使用。