Python:根据键从集合中获取项目

Question

我有一个带有自定义哈希方法的类。

class Test(object):

    def __init__(self, key, value):
        self.key = key # key is unique
        self.value = value

    def __hash__(self):
        # 'value' is unhashable, so return the hash of 'key'
        return hash(self.key)

我使用此类的对象创建了一个集合。

t0, t1, t2 = Test(0, 10), Test(1, 5), Test(2, 10)
s = set([t0, t1, t2])

现在，有什么方法可以使用 key 从 s 中找到对象吗？即我想做:

find_using_key(s, 1) # should return [t1]

我知道我可以通过迭代集合中的项目来做到这一点，但我觉得应该有一个 O(1) 方法来做到这一点，因为 key 有效地确定了“位置” 集合。

Answer 1

... since key effectively determines the 'position' in the set

事实并非如此。具有相同 key 的两个元素可以在集合中共存:

>>> t0, t1 = Test(1,1), Test(1,2)
>>> len(set((t0,t1)))
2

哈希值没有定义相等性。这也是不可能的，因为您可能会发生哈希冲突。

现在关于你的问题:不要使用set。它由具有操作插入 和查找 的抽象接口(interface)定义。它不提供您想要的操作。潜在的底层实现在理论上是否可以支持您想要的操作并不真正相关。相反，使用 dict，并将键与实例相关联。