python - 如何实现类的内置 sum()？

Question

应用于字典的内置函数 sum() 返回其键的总和：

sum({1:0,2:10})=3

我想创建字典的子类，比如 SubDict，并重写一些函数以返回其值的总和，即

sum(SubDict((1,0),(2,10))=10

我需要重写哪个函数来实现这个功能？

ps 这是一个普遍的问题，如何将内置的 sum() 函数实现到一个类中，而不仅仅是在这种特殊情况下。

Answer 1

你可以这样做：

In [1]: sum({1:0,2:10}.values())
Out[1]: 10

如果要实现一个总和sum为值的子类，只需覆盖该__iter__方法：

In [21]: class MyDict(dict):
   ....:     def __iter__(self):
   ....:         for value in self.values():
   ....:             yield value
   ....:

In [22]: d = MyDict({1:0,2:10})

In [23]: sum(d)
Out[23]: 10

但是你将无法做到：

for key in d:
    print d[key]

因为该__iter__函数将返回值...您将始终必须使用该keys()函数：

for key in d.keys():
    print d[key]

更好的解决方案是添加一个sum方法：

In [24]: class MyDict(dict):
   ....:     def sum(self):
   ....:         return sum(self.values())
   ....:

In [25]: d = MyDict({1:0,2:10})

In [26]: d.sum()
Out[26]: 10

Answer 2