Python是否存在可直接用ID操作实例、无需键值对的类字典容器？

为什么直接用foos_set.remove(6)会报错？ #

Python集合（set）查找元素时，先通过哈希值定位候选元素，再用__eq__方法确认元素是否相等。你只重写了__hash__让它基于foo_id，但默认dataclass会自动生成一个比较所有字段的__eq__方法，而且int类型的6和Foo实例不属于同一类型，6 == Foo(6)永远为False，所以集合找不到匹配元素，抛出KeyError。

解决方法 #

方法1：重写__eq__方法，仅比较foo_id #

修改Foo类，让__eq__只校验foo_id，这样无论传入Foo(6)还是直接传6，只要foo_id匹配就能被识别（可按需添加类型判断）：

from dataclasses import dataclass

@dataclass
class Foo:
    foo_id: int
    # 其他字段

    def __hash__(self):
        return self.foo_id.__hash__()
    
    def __eq__(self, other):
        if isinstance(other, int):
            return self.foo_id == other
        elif isinstance(other, Foo):
            return self.foo_id == other.foo_id
        return False

现在foos_set.remove(6)可正常执行，且集合查找依然保持O(1)时间复杂度。

方法2：封装自定义容器（替代字典） #

如果不想修改Foo类，可以自己实现一个简单容器，内部用字典映射foo_id到实例，对外提供类似集合的接口：

class FooSet:
    def __init__(self, foos):
        self._map = {foo.foo_id: foo for foo in foos}
    
    def remove(self, foo_or_id):
        if isinstance(foo_or_id, Foo):
            del self._map[foo_or_id.foo_id]
        else:
            del self._map[foo_or_id]
    
    def add(self, foo):
        self._map[foo.foo_id] = foo
    
    def __contains__(self, foo_or_id):
        if isinstance(foo_or_id, Foo):
            return foo_or_id.foo_id in self._map
        return foo_or_id in self._map
    
    def __iter__(self):
        return iter(self._map.values())

使用示例：

foos_set = FooSet(Foo(i) for i in range(10))
foos_set.remove(6)  # 正常执行

方法3：简化字典写法（其实并不繁琐） #

你提到的字典方案可以简化为一行，操作也很直观：

foos_dict = {foo.foo_id: foo for foo in (Foo(i) for i in range(10))}
# 删除元素
del foos_dict[6]
# 获取实例
target_foo = foos_dict[6]

标准库是否有其他适配容器？ #

Python标准库中没有专门针对“用唯一标识查找实例”场景设计的容器，上述几种方法是最直接的解决方案。

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