Python - 向量数学中的运算符重载

Question

代码:

class Vector:
    def __init__(self):
        self.x = 0.0
        self.y = 0.0

    def __mul__(self, scalar):
        v = Vector()
        v.x = self.x * scalar
        v.y = self.y * scalar
        return v    


v = Vector()
v.x = 2
v.y = 5
v = v * 2

问题:有没有办法以一种可能的方式重载 mul 运算符:

v = 2 * v

？

Answer 1

使用__rmul__ :

def __rmul__(self, scalar):  # self: right operand (Vector)
    v = Vector()
    v.x = self.x * scalar
    v.y = self.y * scalar
    return v    

__rmul__ 当左侧操作数不支持相应操作 (__mul__) 时调用。