← Все вопросы

Как сделать, чтобы объекты можно было складывать через + ?

Задан 4 месяца назад452 просмотров2 ответа
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__.

Ваш ответ

Войдите, чтобы ответить на вопрос.
Поддержать проект