如何生成一个计数器的所有子集？ [英] How to generate all the subsets of a counter?

问题描述

我需要编写一个名为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}]

这篇关于如何生成一个计数器的所有子集？的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！