用户定义的__mul__方法不是可交换的

Question

我编写了一个类来表示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

但这需要重新定义我要与用户定义的类一起使用的每个函数。

有没有办法使内置函数正确处理此问题？

Answer 1

如果要 对不同类型进行 交换 ，则
需要实现__rmul__()。如果实现，则将与所有__r*__()特殊方法一样调用该方法，否则操作将引发a
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() '''