【魔术方法】运算符重载

        python中定义一个类时,可以使用其魔术方法来定制化自己的类,简单来说,魔术方法就是在定义一个类时,用双下划线包起来的方法,现在要说的是其中的一部分,就是运算符重载:

        首先我们要弄清什么是运算符,运算符就是用在变量之间执行某种操作的符号,例如,+、-、*、/、==、<等等符号都属于运算符,在python中重载它们是非常简单的,例如我们现在定义一个Point类,其代表二维笛卡尔坐标系中的一个点

class Point:
    
    def __init__(self, x, y):
        self.x = x
        self.y = y
    
    def __add__(self, other):
        return Point(self.x + other.x, self.y + other.y)
    
    def __sub__(self, other):
        return Point(self.x - other.x, self.y - other.y)

        这样一个简单的Point类就建好了,并且实现了矢量加法与减法,我们还可以实现乘法和除法,但是两个矢量直接相乘或相除其实数学意义是不明确的,因此矢量乘法和除法应该是相对于一个常量而言的

    def __mul__(self, other):
        assert isinstance(other, (int, float))
        return Point(self.x * other, self.y * other)

    def __truediv__(self, other):
        assert isinstance(other, (int, float))
        return Point(self.x / other, self.y / other)

        mul表示乘法,truediv表示浮点除法,也就是 / ,当然还有整数除法floordiv,也就是 // 

        到这里一个Point类的几何信息就基本设计完成了,但这还不够,现在希望可以实现比大小,一个点距离原点的长度比另一个更大就认为该Point大于另一个

from math import sqrt


class Point:

    def __init__(self, x, y):
        self.x = x
        self.y = y
        self.value = sqrt(x ** 2 + y ** 2) # 距离原点的欧氏距离

    def __le__(self, other):
        return self.value <= other.value
    
    def __lt__(self, other):
        return self.value < other.value
    
    def __ge__(self, other):
        return self.value >= other.value
    
    def __gt__(self, other):
        return self.value > other.value
    
    def __eq__(self, other):
        return self.value == other.value
    
    def __ne__(self, other):
        return self.value != other.value

        其实,在这里我们可以不写ne(not equal),ne默认为eq取反,到此为止Point类已经设计完成了,以上也只是非常简单的演示了如何重载运算符,需要注意的是,add、sub、mul、div等等涉及到算术运算的魔术方法都有对应的一个r方法,比如radd、rsub,这是将一个实例对象放在该运算符的右边时对应的处理逻辑,比如想实现Point加一个常数

    def __add__(self, other):
        if isinstance(other, (int, float)):
            return Point(self.x + other, self.y + other)
        return Point(self.x + other.x, self.y + other.y)

p = Point(1, 1)
print(p + 1)    # Point(2, 2)
print(1 + p)    # TypeError: unsupported operand type(s) for +: 'int' and 'Point'

        如果p在+的右侧,执行的是1(int类型)的__add__方法,很显然,int内置的add方法不会知道怎么与Point相加,因此我们需要定义p的__radd__方法,当int尝试调用add失败后,python会继续调用Point的radd方法,只有都失败的情况下才会出现上面的TypeError

    def __radd__(self, other):
        if isinstance(other, (int, float)):
            return Point(self.x + other, self.y + other)
        return Point(self.x + other.x, self.y + other.y)

        这次,上面的两个print就不会报错了

        本来想到此为止的,但最近在学C++,突然想到一个很有意思的东西,就是在python中使用cout,单独开一篇探讨一下

  • 5
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值