将对象转换为字典键

Question

我想知道是否有一种简单的方法可以在字典中为一个值提供多个键。我想要实现的一个例子如下：

class test:
    key="test_key"
    
    def __str__(self):
        return self.key

tester = test()

dictionary = {}
dictionary[tester] = 1

print(dictionary[tester])
print(dictionary["test_key"])

输出将是：

>>> 1
>>> 1

我正在寻找一种在将对象用作键之前自动将其转换为字符串的方法。这可能吗？

Answer 1

几乎可以肯定你不应该这样做；只需使用 dictionary[str(tester)]。它更具可读性，更少惊喜，只需要多写五个字符。

如果你坚持的话，这是我能想到的最好的

class StrKeyedDict(dict):
    def __getitem__(self, key):
        return super().__getitem__(str(key))
    def __setitem__(self, key, value):
        super().__setitem__(str(key), value)
    # ... do all the other methods that mess with the key

class Test:
    key="test_key"

    def __str__(self):
        return self.key

tester = Test()
dictionary = StrKeyedDict()
dictionary[tester] = 1
print(dictionary["test_key"])
# => 1

Answer 2

就我个人而言，我认为将对象显式转换为字符串会更好，例如

dictionary[str(tester)] = 1

也就是说，如果你真的真的 REALLY 确定要这样做，请定义 __hash__ 和 __eq__ dunder 方法。无需创建新的数据结构或更改 class 定义之外的现有代码：

class test:
    key="test_key"
    
    def __hash__(self):
        return hash(self.key)
        
    def __eq__(self, other):
        if isinstance(other, str):
            return self.key == other
        return self.key == other.key
    
    def __str__(self):
        return self.key

这将输出：

1
1

将对象转换为字典键

Converting object to dictionary key

python

string

dictionary

object

hashmap