如何在frozenset集合中筛选出所有为其他frozenset子集的元素

嘿，我来帮你搞定这个问题！你需要的是从包含多个frozenset的集合里，筛选出所有是其他frozenset子集的元素对吧？咱们一步步实现这个get_subsets函数。

核心思路 #

对于集合里的每个frozenset s，我们只需要确认：原集合中存在另一个和它不同的frozenset t，满足 s 是 t 的子集（也就是 s.issubset(t) 为True）。只要满足这个条件，s 就是我们要找的目标元素。

基础实现代码 #

先写一个直观的版本，逻辑清晰容易理解：

def get_subsets(my_set):
    subsets = set()
    # 遍历每个待检查的frozenset
    for s in my_set:
        # 检查集合里的其他元素是否包含当前s
        for t in my_set:
            if s != t and s.issubset(t):
                subsets.add(s)
                break  # 找到一个符合条件的t就不用继续检查了
    return subsets

测试你的示例 #

用你给出的集合测试一下：

my_set = { frozenset({1, 2, 3}), frozenset({1, 2, 3, 4}), frozenset({1, 2, 3, 5}), frozenset({1, 3, 5}), frozenset({1, 2}) }
subsets = get_subsets(my_set)
print(subsets)

输出结果会是：

{frozenset({1, 2}), frozenset({1, 2, 3}), frozenset({1, 3, 5})}

咱们验证一下：

• frozenset({1,2}) 是 {1,2,3}、{1,2,3,4}、{1,2,3,5} 的子集
• frozenset({1,2,3}) 是 {1,2,3,4} 和 {1,2,3,5} 的子集
• frozenset({1,3,5}) 是 {1,2,3,5} 的子集
• 剩下的 {1,2,3,4} 和 {1,2,3,5} 没有被任何其他集合包含，所以不在结果里。

优化版本（适合大集合） #

如果你的集合里元素很多，上面的双重遍历效率会有点低。我们可以先按集合的大小从小到大排序，因为小集合不可能被更小的集合包含，所以只需要检查后面更大的集合：

def get_subsets_optimized(my_set):
    # 按集合元素数量从小到大排序
    sorted_sets = sorted(my_set, key=lambda x: len(x))
    subsets = set()
    for i in range(len(sorted_sets)):
        current_set = sorted_sets[i]
        # 只检查后面比当前集合大的元素
        for j in range(i + 1, len(sorted_sets)):
            candidate_set = sorted_sets[j]
            if current_set.issubset(candidate_set):
                subsets.add(current_set)
                break
    return subsets

这个版本能减少不必要的遍历，处理大集合时速度会更快。

内容的提问来源于stack exchange，提问作者alliedtoasters