← Все вопросы
Как сделать, чтобы объекты можно было складывать через + ?
14
Хочу свой класс вектора, чтобы v1 + v2 возвращал новый вектор с суммой координат. Сейчас выдаёт TypeError: unsupported operand type(s) for +. Что нужно добавить?
2 ответа
22
✓ Принятый ответ — помог автору
Определи метод __add__:
class Vector:
def __init__(self, x, y):
self.x = x
self.y = y
def __add__(self, other):
return Vector(self.x + other.x, self.y + other.y)
def __repr__(self):
return f'Vector({self.x}, {self.y})'
print(Vector(1, 2) + Vector(3, 4)) # Vector(4, 6)
Когда пишешь a + b, Python вызывает a.__add__(b). Важно: возвращай новый объект, а не меняй self на месте — так ведут себя обычные числа. Аналогично есть __sub__ (-), __mul__ (*), __truediv__ (/). А если хочешь поддержать 2 * v (число слева), нужен ещё __rmul__.
Софья Баранова Про возврат нового объекта — частая ошибка новичков, они мутируют self и удивляются · 3 месяца назад
5
Метод __add__.
Ваш ответ
Войдите, чтобы ответить на вопрос.