如何提升嵌套frozenset的打印可读性？

当打印包含frozenset的集合时，输出会被大量frozenset({})标识掩盖，可读性极差：

A = frozenset(["A1", "A2"])
B = frozenset(["B1", "B2"])
C = frozenset(["C1", "C2"])

foo = {A, B, C}

print(foo)
# {frozenset({'B1', 'B2'}), frozenset({'C1', 'C2'}), frozenset({'A1', 'A2'})}

print(*foo)
# frozenset({'B1', 'B2'}) frozenset({'C1', 'C2'}) frozenset({'A1', 'A2'})

需求是隐藏数据集合类型的打印标识，仅关注条目分组，理想输出为：

({'B1', 'B2'}, {'C1', 'C2'}, {'A1', 'A2'})

背景：在寻找大量元素的最优分组时，get_best_grouping函数返回包含frozenset的集合，需要快速验证分组是否合理，且不希望修改原生类型的str方法。

解决方案 #

方法1：转换为普通集合后格式化输出 #

直接将每个frozenset转为普通set，再用元组格式化输出：

foo = {A, B, C}
print(tuple(set(group) for group in foo))
# ({'B1', 'B2'}, {'C1', 'C2'}, {'A1', 'A2'})

如果要严格匹配理想格式，可手动拼接字符串：

print(f"({', '.join(str(set(g)) for g in foo)})")
# ({'B1', 'B2'}, {'C1', 'C2'}, {'A1', 'A2'})

方法2：自定义打印函数（复用性强） #

如果需要多次验证分组，写一个简单的打印函数：

def print_groups(groups):
    formatted_groups = ', '.join(str(set(g)) for g in groups)
    print(f"({formatted_groups})")

# 调用示例
print_groups(foo)
# ({'B1', 'B2'}, {'C1', 'C2'}, {'A1', 'A2'})

# 结合业务代码使用
items = {'B1', 'B2', 'C1', 'C2', 'A1', 'A2'}
pairs = get_best_grouping(items)
print_groups(pairs)

方法3：字符串替换（临时调试用） #

临时调试时，直接对原输出做字符串替换最省事：

# 一步替换掉类型标识
print(str(foo).replace('frozenset({', '{').replace('})', '}'))
# ({'B1', 'B2'}, {'C1', 'C2'}, {'A1', 'A2'})

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