python's sum() and non-integer values
Its a bit tricky - the sum() function takes the start and adds it to the next and so on
You need to implement the __radd__
method:
class T:
def __init__(self,x):
self.x = x
def __radd__(self, other):
return other + self.x
test = (T(1),T(2),T(3),200)
print sum(test)
You may also need to implement the __radd__
function, which represents "reverse add" and is called when the arguments can't be resolved in the "forward" direction. For example, x + y
is evaluated as x.__add__(y)
if possible, but if that doesn't exist then Python tries y.__radd__(x)
.
Since the sum()
function starts with the integer 0
, the first thing it does is try to evaluate:
0 + Foo(3)
which will require that you implement Foo.__radd__
.
Try:
import operator
result=reduce(operator.add, mylist)
sum() works probably faster, but it is specialized for builtin numbers only. Of course you still have to provide a method to add your Foo() objects. So full example:
class Foo(object):
def __init__(self, i): self.i = i
def __add__(self, other):
if isinstance(other, int):
return Foo(self.i + other)
return Foo(self.i + other.i)
def __radd__(self, other):
return self.__add__(other)
import operator
mylist = [Foo(42), Foo(36), Foo(12), 177, Foo(11)]
print reduce(operator.add, mylist).i