Определенный пользователем метод __mul__ не является коммутативным

Я написал класс для представления векторов в Python (в качестве упражнения), и у меня возникли проблемы с расширением встроенных операторов.

Я определил метод __mul__ для векторного класса. Проблема в том, что в выражении x * y интерпретатор вызывает метод __mul__ x, а не y.

Итак, vector(1, 2, 3) * 2 возвращает вектор ‹2, 4, 6>, как и должно быть; но 2 * vector(1, 2, 3) создает TypeError, потому что встроенный класс int не поддерживает умножение на мои пользовательские векторы.

Я мог бы решить эту проблему, просто написав новую функцию умножения

def multiply(a, b):
    try:
        return a * b
    except TypeError:
        return b * a

но это потребует переопределения каждой функции, которую я хочу использовать с моими пользовательскими классами.

Есть ли способ заставить встроенную функцию правильно обрабатывать это?


python extend built-in
person smackcrane    schedule 20.08.2011    source источник

Ответы (2)


arrow_upward
32
arrow_downward

Если вам нужна коммутативность для разных типов, вам необходимо реализовать __rmul__() . Если он реализован, он вызывается, как и все специальные методы __r*__(), если в противном случае операция вызвала бы TypeError. Помните, что аргументы меняются местами:

class Foo(object):
    def __mul_(self, other):
        ''' multiply self with other, e.g. Foo() * 7 '''
    def __rmul__(self, other):
        ''' multiply other with self, e.g. 7 * Foo() '''
person pillmuncher    schedule 20.08.2011
comment
Откуда Python знает, что нужно вызывать Foo.__rmul__, а не __mul__ другого объекта? - person ; 20.08.2011
comment
Он пытается использовать __mul__ в левой части, и если не может его найти, то ищет __rmul__ в правой части. - person Karl Knechtel; 20.08.2011
comment
@pst, Python сначала вызывает __mul__, но если он либо не существует, либо возвращает NotImplemented, тогда Python вызывает __rmul__ другого объекта. (Обратите внимание, что если __mul__ вызывает исключение, __rmul__ не вызывается. - person Ethan Furman; 20.08.2011

arrow_upward
3
arrow_downward

Я полагаю, что вы ищете __rmul__

person Michael Kent    schedule 20.08.2011