Подтвердить что ты не робот

Переопределение "+ =" в Python? (__iadd __())

Можно ли переопределить + = в Python?

4b9b3361

ОТВЕТЫ

Ответ 1

Да, переопределите метод __iadd__. Пример:

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

Ответ 2

В дополнение к перегрузке __iadd__ (не забудьте вернуть self!), вы также можете отступить на __add__, так как x + = y будет работать как x = x + y. (Это одна из ловушек оператора + =.)

>>> 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

Он даже отправляет экспертов:

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

x = Resource()
y = Resource()

Какие значения вы ожидаете x.id, y.id и Resource.class_counter?

Ответ 3

В дополнение к тому, что правильно указано в ответах выше, стоит явно разъяснить, что когда __iadd__ переопределяется, операция x += y НЕ заканчивается с конце метода __iadd__.

Вместо этого он заканчивается на x = x.__iadd__(y). Другими словами, Python присваивает возвращаемое значение вашей реализации __iadd__ объекту, который вы "добавляете", ПОСЛЕ завершения реализации.

Это означает, что можно изменить левую часть операции x += y, чтобы завершился неявный шаг. Посмотрите, что может случиться, когда вы добавляете что-то, что находится в списке:

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

Теперь, если ваша реализация __iadd__ (метод объекта at x[1]) ошибочно или намеренно удаляет первый элемент (x[0]) с самого начала списка, Python будет запускать ваш __iadd__ метод) и попытайтесь присвоить его возвращаемое значение x[1]. Который больше не будет существовать (он будет в x[0]), в результате получится ÌndexError.

Или, если ваш __iadd__ вставляет что-то в начало x из приведенного выше примера, ваш объект будет находиться в x[2], а не x[1], а все, что было ранее в x[0], теперь будет на x[1] и назначить возвращаемое значение вызова __iadd__.

Если кто-то не поймет, что происходит, возникающие ошибки могут стать кошмаром для исправления.