Python：如何创建本质为dict、可按set操作且自定义__str__的结构？

要满足你的需求，核心思路是创建一个包装类：内部维护dict[str, list[str]]结构，同时实现set的核心操作接口（比如遍历、成员判断、添加元素），并且自定义__str__方法来输出指定格式的dict内容。如果需要把这个结构作为类属性管理，可以结合描述符返回包装类实例，而非直接返回列表。

基础包装类实现 #

这个类内部管理目标dict，对外完全像set一样操作，打印时输出自定义格式的dict：

class DictBackedSet:
    def __init__(self, initial_dict=None):
        # 初始化内部存储的dict结构
        self._internal_dict = initial_dict.copy() if initial_dict else {}

    # 实现set的遍历接口：迭代所有元素
    def __iter__(self):
        for value_list in self._internal_dict.values():
            yield from value_list

    # 实现set的成员判断：检查元素是否存在于内部dict的任意列表中
    def __contains__(self, item):
        return any(item in value_list for value_list in self._internal_dict.values())

    # 实现set的长度统计：返回所有元素的总数量
    def __len__(self):
        return sum(len(v) for v in self._internal_dict.values())

    # 实现set的添加操作：这里需要你根据实际规则把元素分配到对应的key下
    # 示例规则：按元素首字母小写作为key
    def add(self, item):
        key = item[0].lower()
        if key not in self._internal_dict:
            self._internal_dict[key] = []
        # 避免重复添加
        if item not in self._internal_dict[key]:
            self._internal_dict[key].append(item)

    # 实现set的批量添加操作
    def update(self, items):
        for item in items:
            self.add(item)

    # 自定义打印格式：输出排序后的dict结构
    def __str__(self):
        # 这里可以根据需求调整格式，比如对列表排序、美化输出
        sorted_dict = {k: sorted(v) for k, v in self._internal_dict.items()}
        return str(sorted_dict)

    # 可选：提供获取内部原始dict的方法
    def get_raw_dict(self):
        return self._internal_dict.copy()

使用示例 #

class Foo:
    def __init__(self):
        self.my_dict = DictBackedSet()

# 测试操作
foo = Foo()
foo.my_dict.add("Apple")
foo.my_dict.add("Banana")
foo.my_dict.add("Apricot")
foo.my_dict.update({"Cherry", "Date"})

# 像set一样判断成员
print("Apple" in foo.my_dict)  # 输出 True
# 像set一样遍历
for item in foo.my_dict:
    print(item)  # 输出 Apple、Apricot、Banana、Cherry、Date
# 打印时输出自定义格式的dict
print(foo.my_dict)  # 输出 {'a': ['Apple', 'Apricot'], 'b': ['Banana'], 'c': ['Cherry'], 'd': ['Date']}

结合描述符实现类属性管理 #

如果需要把这个结构作为类属性（而非实例属性）统一管理，可以用描述符返回包装类实例，这样既保留类属性的特性，又能正常使用set操作和自定义打印：

class DictBackedSetDescriptor:
    def __set_name__(self, owner, name):
        self.name = name

    def __get__(self, obj, objtype=None):
        # 类层面访问时返回描述符自身，实例层面访问时返回包装类实例
        if obj is None:
            return self
        # 懒初始化：实例第一次访问时创建包装类对象
        if self.name not in obj.__dict__:
            obj.__dict__[self.name] = DictBackedSet()
        return obj.__dict__[self.name]

    def __set__(self, obj, value):
        # 支持直接传入set或dict来赋值
        if isinstance(value, set):
            dbs = DictBackedSet()
            dbs.update(value)
            obj.__dict__[self.name] = dbs
        elif isinstance(value, dict):
            obj.__dict__[self.name] = DictBackedSet(value)
        else:
            raise TypeError("赋值仅支持set或dict类型")

# 使用描述符的类
class Foo:
    my_dict = DictBackedSetDescriptor()

# 测试类属性操作
foo1 = Foo()
foo1.my_dict.add("Apple")
print(foo1.my_dict)  # 输出 {'a': ['Apple']}

foo2 = Foo()
foo2.my_dict = {"b": ["Banana"], "c": ["Cherry"]}
print(foo2.my_dict)  # 输出 {'b': ['Banana'], 'c': ['Cherry']}

关键说明 #

• 之前的描述符方案问题在于__get__返回的是列表，无法自定义__str__；而包装类自身控制__str__，同时提供set的操作接口，完美匹配需求。
• 添加元素的规则（比如如何把set元素映射到dict的key）需要你根据实际业务调整add方法中的逻辑。

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