Python:列表中每个可能的子集组合

问题描述 投票:0回答:1

我有一个清单:

list = ["M","M","M","C","C","C"]

我希望输出为

out = [["M"],["C"],["M,"C"],["M","M"],["C","C"]]

如何执行此操作?预先感谢!

我已使用此代码:

from itertools import chain, combinations
stuff = ["M","M","M","C","C","C"]
def all_subsets(ss):
    p = map(lambda x: combinations(ss, x), range(1, 3))
    return p

for subset in all_subsets(stuff):
    print(list(subset))

for subset in all_subsets(stuff):
    print(list(subset))

但是输出是enter image description here

python algorithm combinations itertools
1个回答
1
投票

如果只需要1和2个元素的组合,则可以尝试itertools.combinations

>>> from itertools import combinations
>>> l = ["M", "M", "M", "C", "C", "C"]
>>> combs = set(list(combinations(l, 1)) + list(combinations(l, 2)))
>>> out = list(map(list, combs))
>>> print(out)
[['C', 'C'], ['M', 'M'], ['C'], ['M', 'C'], ['M']]
© www.soinside.com 2019 - 2024. All rights reserved.