运算符重载

2020年6月25日 2018年7月22日
运算符重载

运算符重载

Python 中的一切都是对象,每个对象都有一些特殊的内部方法,用于与其他对象进行交互。通常,这些方法遵循 __action__ 命名约定。

你可以重载这些方法中的任何一种。这通常用于 Python 中的运算符重载。下面是使用 Python 数据模型进行运算符重载的示例。在 Vector 类创建有两个变量的简单矢量。我们将使用运算符重载来支持两个向量的数学运算。

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

    def __add__(self, v):
        # Addition with another vector.
        return Vector(self.x + v.x, self.y + v.y)

    def __sub__(self, v):
        # Subtraction with another vector.
        return Vector(self.x - v.x, self.y - v.y)

    def __mul__(self, s):
        # Multiplication with a scalar.
        return Vector(self.x * s, self.y * s)

    def __div__(self, s):
        # Division with a scalar.
        float_s = float(s)
        return Vector(self.x / float_s, self.y / float_s)

    def __floordiv__(self, s):
        # Division with a scalar (value floored).
        return Vector(self.x // s, self.y // s)

    def __repr__(self):
        # Print friendly representation of Vector class. Else, it would
        # show up like, <__main__.Vector instance at 0x01DDDDC8>.
        return '<Vector (%f, %f)>' % (self.x, self.y, )

a = Vector(3, 5)
b = Vector(2, 7)

print a + b # Output: <Vector (5.000000, 12.000000)>
print b - a # Output: <Vector (-1.000000, 2.000000)>
print b * 1.3 # Output: <Vector (2.600000, 9.100000)>
print a // 17 # Output: <Vector (0.000000, 0.000000)>
print a / 17 # Output: <Vector (0.176471, 0.294118)>

通过运算符重载,我们可以让类进行一些数学运算,假如这些运算经常被调用到的话,这样就省去了通过函数来实现的步骤。

具体的可支持重载的运算符列表可以参见官方手册