Курс Python → Метод rsub для пользовательских чисел

Метод rsub в Python предоставляет возможность расширить функциональность вычитания для пользовательских объектов. Этот метод используется в случае, когда операнд слева не имеет поддержки для метода sub, и позволяет определить, как ваш объект должен вычитаться из другого объекта. Это полезно, когда вам необходимо определить поведение вычитания для вашего пользовательского класса.

Для демонстрации этого принципа, предположим, что у нас есть класс, который представляет пользовательские числа. Мы хотим, чтобы наш класс поддерживал операцию вычитания как с обычными числами Python (например, int, float), так и с другими экземплярами нашего класса. Используя метод rsub, мы можем определить, как должно происходить вычитание для наших пользовательских чисел, обеспечивая удобство использования и гибкость класса.

Пример кода ниже показывает, как можно реализовать метод rsub для пользовательского класса, который представляет пользовательские числа:


class CustomNumber:
    def __init__(self, value):
        self.value = value

    def __rsub__(self, other):
        if isinstance(other, (int, float)):
            return other - self.value
        elif isinstance(other, CustomNumber):
            return other.value - self.value
        else:
            raise TypeError("Unsupported operand type for rsub")

В этом примере метод __rsub__ определяет поведение вычитания для пользовательского класса CustomNumber. Если операнд слева является целым числом или числом с плавающей запятой, он вычитает значение объекта CustomNumber из переданного числа. Если операнд слева является другим экземпляром класса CustomNumber, метод также выполняет вычитание. Если тип операнда не поддерживается, возникает исключение TypeError.

Твои коллеги будут рады, поделись в