Курс Python → Переопределение метода pow

Python предоставляет возможность переопределить метод pow() для объектов пользовательских классов. Этот лайфхак полезен, когда вы работаете с комплексными числами или другими типами данных, где нужно определить операции возведения в степень и взятия по модулю. Переопределение метода pow() позволяет вашему объекту поддерживать стандартные арифметические операции в Python.

Для переопределения метода pow() в вашем классе, вам нужно определить метод __pow__. Этот метод принимает два аргумента: первый — объект, который нужно возвести в степень, и второй — необязательный аргумент, который определяет модуль. Возвращаемое значение метода __pow__ должно быть результатом операции возведения в степень или взятия по модулю.


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

    def __pow__(self, exponent, modulus=None):
        if modulus is None:
            return self.value ** exponent
        else:
            return pow(self.value, exponent, modulus)

В приведенном выше примере, мы создаем пользовательский класс CustomNumber, который имеет метод __pow__ для поддержки операций возведения в степень и взятия по модулю. Мы можем использовать этот класс для работы с числами и выполнять арифметические операции с помощью стандартных операторов Python.

Используя переопределение метода pow(), вы можете создавать объекты, которые поддерживают арифметические операции возведения в степень и взятия по модулю. Это удобно при работе с различными типами данных и позволяет вам определить поведение вашего объекта при выполнении арифметических операций в Python.

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