python怎么实现sum叠加_python – 如何实现类的内置sum()？

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

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

我想创建字典的子类,比如说SubDict,并覆盖一些函数来返回其值的总和,即

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

我需要覆盖哪个功能才能实现此功能？

附：这是一个普遍的问题,如何在类中实现内置的sum()函数,而不仅仅是在这种特殊情况下.

解决方法:

你可以这样做：

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]

更好的解决方案是添加求和方法：

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

标签：python