一个python类，是项目的1-1映射，继承自dict

Question

我从 dict 类继承并尝试对项目进行 1 对 1 映射。

我正在尝试实现给定的 setitem 而不是 key:value key1:key 2, key2:key1 也被添加。 如果其中任何一个键已经存在，则应在添加对之前弹出它们。我该如何实现 setitem？

init 应该能够构造一个空映射或来自给定字典的映射

有什么想法吗？谢谢

class Mapping(dict):
    def __init__(self, Map = None):
        if Map = None:
            Map = dict()
        else:
            self.update(Map)

    def __repr__(self):
        return "Mapping({})".format(dict.__repr__self)
    def pop(self):
        return self.pop   #??? How can I pop a key and also pop the key it maps to?
    def __setitem__(self, key1,key2):
        self.__dict__[key1] = key2
        self.__dict__[key2] = key1

Answer 1

pop 将键作为参数并返回与其关联的值。因此，将您从弹出 key1 中获得的值作为 key2 弹出。

你可以self.__dict__.pop(self.__dict__.pop(key))

Answer 2

你可以尝试做这样的事情：

from collections.abc import MutableMapping

# We need to implement those method: __getitem__, __setitem__, __delitem__, __iter__, __len__

class TwoWayDict(MutableMapping):

    __slots__ = '__dict__'

    def __init__(self, *args, **kwargs):
        self.__dict__.update(*args, **kwargs)
    def __getitem__(self, key):
        return self.__dict__[key]
    def __setitem__(self, key, value):
        self.__dict__[key] = value
        self.__dict__[value] = key
    def __delitem__(self, key):
        self.__dict__.__delitem__(self[self[key]])            
        self.__dict__.__delitem__(self[key])  # this do not work if key == value
    def __iter__(self):
        return iter(self.__dict__)
    def __len__(self):
        return len(self.__dict__)
    def get(self, key, default=None):
        return self[key] if key in self else default
    def __str__(self):
        return str(self.__dict__)

测试：

a = TwoWayDict()
a['key1'] = 'key2'
print(a)
print(a['key1'])
print(a['key2'])
print(len(a))
for key, val in a.items():
    print(key, val)

结果：

{'key1': 'key2', 'key2': 'key1'}
key2
key1
2
key1 key2
key2 key1

请注意，长度和迭代可能不明确。如果您不想显示重复键，则需要在 dict 项中找到键并返回相应的键。对于流行音乐，您可以使用 SW Williams 提出的解决方案。

编辑：如果你真的需要从 dict 继承（可能是一个赋值......）不要忘记实现更新，因为它会绕过 __setitem__ 并进入 dict。此外，如果您有一个 1 对 1 的键映射，您需要在 __del__ 和 __pop__ 函数中处理这个问题。