Python求多集合专属交集：如何实现更简洁高效的方案？

问题说明 #

给定示例集合：

set1 = {"aa","ab","ba","cc","ca"}
set2 = {"fr","fa","po","pl","cc"}
set3 = {"fr","xz","hn","fa"}
set4 = {"jq","we","hn","ca","aa","fk"}
set5 = {"jp","wx","we","fr","ba"}

我们需要计算专属交集：即某几个集合的交集中，必须排除那些同时属于更大集合交集的元素。比如set2与set3的原始交集是{"fr","fa"}，但"fr"同时属于set2、set3、set5的交集，因此二者的专属交集仅为{"fa"}。

现有方法的问题 #

手动逐步减法的方式需要逐个计算不同大小子集的交集，再手动减去所有更大子集的交集，不仅代码繁琐冗余，而且当集合数量增加时，维护成本极高，完全不具备扩展性。

简洁实现方案 #

利用itertools.combinations生成所有子集，按子集大小从大到小处理，自动完成"减去更大子集专属交集"的操作，代码如下：

import itertools

# 定义示例集合
set1 = {"aa","ab","ba","cc","ca"}
set2 = {"fr","fa","po","pl","cc"}
set3 = {"fr","xz","hn","fa"}
set4 = {"jq","we","hn","ca","aa","fk"}
set5 = {"jp","wx","we","fr","ba"}
set_list = [set1, set2, set3, set4, set5]

# 用字典存储每个子集的专属交集，键为集合索引的元组（如(1,2)对应set2和set3）
exclusive_isects = {}

# 从最大子集（所有集合）开始，逐步处理到2个集合的子集
for subset_size in range(len(set_list), 1, -1):
    # 生成当前大小的所有集合索引组合
    for subset_indices in itertools.combinations(range(len(set_list)), subset_size):
        # 计算当前子集的原始交集
        current_isect = set.intersection(*[set_list[i] for i in subset_indices])
        # 减去所有包含当前子集的更大子集的专属交集
        remaining_indices = [i for i in range(len(set_list)) if i not in subset_indices]
        for add_indices in itertools.combinations(remaining_indices, 1):
            super_subset = subset_indices + add_indices
            if super_subset in exclusive_isects:
                current_isect -= exclusive_isects[super_subset]
        exclusive_isects[subset_indices] = current_isect

# 示例：查看set2与set3的专属交集（对应索引1、2）
print("set2与set3的专属交集:", exclusive_isects[(1, 2)])

# 遍历输出所有非空的专属交集
print("\n所有非空专属交集：")
for indices, isect in exclusive_isects.items():
    if isect:
        set_labels = [f"set{i+1}" for i in indices]
        print(f"{', '.join(set_labels)}: {isect}")

代码逻辑说明 #

1. 从大到小处理子集：先计算最大子集（所有集合）的专属交集（即它本身的交集，无更大子集可减），再依次处理更小的子集。
2. 自动减去更大子集的结果：对于每个子集，找到所有包含它的更大子集（比当前子集多一个集合的组合），用当前子集的原始交集减去这些更大子集的专属交集，得到当前子集的专属结果。
3. 通用性强：不管集合数量是多少，代码都能自动处理所有可能的子集组合，无需手动修改。

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