Przesłanianie "+ =" w Pythonie? (__iadd __() metoda)

Question

Czy można zastąpić + = w Pythonie?

Answer 1

Tak, należy zastąpić metodę __iadd__. Przykład:

def __iadd__(self, other): 
    self.number += other.number 
    return self  

Answer 2

http://docs.python.org/reference/datamodel.html#emulating-numeric-types

Na przykład, w celu wykonania instrukcji x + = y, gdzie x jest wystąpienie klasy, posiada metody __iadd __(), x .__ iadd__ (y) jest wywoływane.

Answer 3

Poza przeciążeniem __iadd__ (pamiętaj, aby wrócić siebie!), Można również awaryjne na __add__, a x + y = zadziała jak x = x + y. (Jest to jedna z pułapek operatora + =.)

>>> class A(object): 
... def __init__(self, x): 
...  self.x = x 
... def __add__(self, other): 
...  return A(self.x + other.x) 
>>> a = A(42) 
>>> b = A(3) 
>>> print a.x, b.x 
42 3 
>>> old_id = id(a) 
>>> a += b 
>>> print a.x 
45 
>>> print old_id == id(a) 
False 

nawet trips up experts:

class Resource(object): 
    class_counter = 0 
    def __init__(self): 
    self.id = self.class_counter 
    self.class_counter += 1 

x = Resource() 
y = Resource() 

Jakie wartości można się spodziewać x.id, y.id i Resource.class_counter mieć?

Answer 4

Oprócz co prawidłowo podano w odpowiedzi powyżej, warto wyraźnie wyjaśnić, że kiedy __iadd__ jest nadpisane, operacja x += y nie kończy się wraz z końcem __iadd__ metody.

Zamiast tego kończy się x = x.__iadd__(y). Innymi słowy, Python przypisuje wartość zwracaną przez implementację do obiektu, do którego "dodajesz", PO zakończeniu implementacji.

Oznacza to, że możliwe jest zmutowanie lewej strony operacji x += y, aby nie zakończył się ostatni niejawny krok. Zastanów się, co może się zdarzyć podczas dodawania do czegoś, co jest w liście:

>>> x[1] += y # x has two items

Teraz, jeśli realizacja __iadd__ (metoda obiektu w x[1]) omyłkowo lub celowo usuwa pierwszy element (x[0]) od początku listy, Python uruchomi następnie twoją metodę __iadd__) spróbuj przypisać jej wartość zwracaną do x[1]. Który już nie będzie istnieć (będzie to x[0]), w wyniku czego powstanie ÌndexError.

Lub, jeśli __iadd__ wkładki coś do początku x powyższego przykładu, Twój obiekt będzie na x[2], nie x[1], a co było wcześniej w x[0] będzie teraz w x[1] i przypisać wartość zwracaną przez __iadd__ wezwanie.

O ile nie wiadomo, co się dzieje, powstałe błędy mogą być koszmarem do naprawienia.