python – 与dict.fromkeys()和类似dict的对象的KeyError

2023年11月28日 280次阅读

在
Python中,您可以使用字典作为dict.fromkeys()的第一个参数,例如：

In [1]: d = {'a': 1, 'b': 2}

In [2]: dict.fromkeys(d)
Out[2]: {'a': None, 'b': None}

我尝试用类似dict的对象做同样的事情,但这总是引发一个KeyError,例如：

In [1]: class SemiDict:
   ...:     def __init__(self):
   ...:         self.d = {}
   ...:
   ...:     def __getitem__(self, key):
   ...:         return self.d[key]
   ...:
   ...:     def __setitem__(self, key, value):
   ...:         self.d[key] = value
   ...:
   ...:

In [2]: sd = SemiDict()

In [3]: sd['a'] = 1

In [4]: dict.fromkeys(sd)
---------------------------------------------------------------------------
KeyError                                  Traceback (most recent call last)

C:\bin\Console2\<ipython console> in <module>()

C:\bin\Console2\<ipython console> in __getitem__(self, key)

KeyError: 0

到底发生了什么？除了使用像dict.fromkeys(sd.d)这样的东西之外,还能解决吗？

最佳答案 要创建dict,fromkeys会遍历其参数.所以它必须是一个迭代器.使其工作的一种方法是在你的dict中添加一个__iter__方法：

def __iter__(self):
    return iter(self.d)
点赞