如何生成一个计数器的所有子集？

本文介绍了如何生成一个计数器的所有子集？的处理方法，对大家解决问题具有一定的参考价值，需要的朋友们下面随着小编来一起学习吧！ 问题描述

我需要编写一个名为char_counts_subsets的函数，该函数将字符计数的字典作为参数，并返回该字典的所有子集，考虑到字符计数的值。示例代码如下所示：

char_counts = {"a": 1, "b": 2} def char_counts_subsets(cc): return [{}, {"b": 1}, {"b": 2}, {"a": 1}, {"a": 1, "b": 1}, {"a": 1, "b": 2} ] # ordering of the subsets isn't important print(char_counts_subsets(char_counts))

如何泛化此函数以使其适用于任何cc词典？

推荐答案

这是一个组合问题，最好使用itertools来解决。

from itertools import product

将每个词典项展开为一系列项：

range_items = [[(x, z) for z in range(y + 1)] for x,y in char_counts.items()] #[[('a', 0), ('a', 1)], [('b', 0), ('b', 1), ('b', 2)]]

取每个范围中的每个项目与所有其他范围中的每个项目的笛卡尔乘积：

products = product(*range_items) #[(('a', 0), ('b', 0)), (('a', 0), ('b', 1)),...(('a', 1), ('b', 2))]

剔除有0个计数器的对，并将剩余部分转换为具有词典理解的词典：

[{k: v for k, v in pairs if v > 0} for pairs in products] #[{}, {'b': 1}, {'b': 2}, {'a': 1}, {'a': 1, 'b': 1}, {'a': 1, 'b': 2}]