operator overload python custom class

2019-06-11 03:50发布

问题:

say I want to overload an operator (lets say + for now) for some class I've created,

class A (object):
    #code here

and then:

a = A()
b = A()

what would I do to define:

c = a + b

or something along those lines? (note: this question is purely theoretical I will likely use this at some time, just not currently (unless its really easy and I really need a use for it))

(p.s. if it is possible to do this for other things such as and, or, not, str, e.t.c.)

回答1:

Since A is the first operand of the binary operator, you should define A.__add__(). If A was the second operand and there was no way to change the class of the first operand, you would have to define A.__radd__().

Full reference for magic methods