Python魔法函数(特殊函数)

liftword3周前 (06-07)技术文章4

Python中如何实现运算符的重载,即实现例如a+b这样的运算符操作呢?

在C++中可以使用operator关键字实现运算符的重载。但是在Python中没有类似这样的关键字,所以要实现运算符的重载,就要用到Python的魔法函数。Python魔法函数是以双下划线开头,双下划线结尾的一组函数。我们在类定义中最常用到的__init__函数就是这样一个魔法函数,它在创建类对象时被自动调用。

下面我们来看个简单的例子。

Bash
#定义一个三维向量,实现加减等基本运算
class Vector3:
    def __init__(self,x,y,z):
        self.x = x
        self.y = y
        self.z = z
    
    # 重载 '+' 二元运算符(a+b)
    def __add__(self, other):
        return Vector3(self.x+other.x,self.y+other.y,self.z+other.z)
    
    # 重载 '-' 二元运算符(a-b)
    def __sub__(self, other):
        return Vector3(self.x-other.x,self.y-other.y,self.z-other.z)
    
    # 重载 '+=' 二元运算符 (a+=b)
    def __iadd__(self, other):
        self.x+=other.x
        self.y+=other.y
        self.z+=other.z
        return self
    
    # 重载 '-=' 二元运算符 (a-=b)
    def __isub__(self, other):
        self.x-=other.x
        self.y-=other.y
        self.z-=other.z
        return self
    
    ## 重载 '-' 一元运算符 (-a)
    def __neg__(self):
        return Vector3(-self.x,-self.y,-self.z)
    
    #重载 str()
    def __str__(self):
        return "[{},{},{}]".format(self.x,self.y,self.z)
    
    #重载 操作符'()'
    def __call__(self):
        print("{}({}, {}, {})".format(self.__class__.__name__,self.x,self.y,self.z))

v1 = Vector3(1,5,6)
v2 = Vector3(3,2,1)
v3 = Vector3(-1,0,1)

print("v1+v2 =",v1+v2) #输出结果: v1+v2= [4,7,7]
print("v1+v2+v3 =",v1+v2+v3) #输出结果: v1+v2+v3 = [3,7,8]
print("v1-v2 =",v1-v2) #输出结果: v1-v2= [-2,3,5]

v1+=v2
print("v1+=v2; v1 =",v1) #输出结果: v1+=v2; v1 = [4,7,7]

v1-=v2
print("v1-=v2; v1 =",v1) #输出结果: v1-=v2; v1 = [1,5,6]

print("-v1 =", -v1)   #输出结果: -v1 = [-1,-5,-6]

v1() #输出结果: Vector3(1, 5, 6)

上述代码示例了几个魔法函数的用法。__add__函数对应了二元运算符+,当执行a+b语句时,python就会自动调用a.add(b)。 对于上述例子中的v1+v2+v3,则相当于调用了(v1.add(v2)).add(v3)。

代码中还有一个在Python类定义经常使用的__str__函数,当使用str()时会被调用。print函数对传入的参数都调用了str()将其转换成易读的字符串形式,便于打印输出,因而会调用类定义的__str__函数打出自定义的字符串。

代码中还有一个特殊的__call__函数,该函数在将对象采用函数调用方式使用时被调用, 例如v1()相当于v1.call()。

以上就是魔法函数的基本使用方法。常见的魔法函数我们可以使用dir()函数来查看。

Bash
dir(v1)

输出结果为:

Bash
['__add__',
'__call__',
'__class__',
'__delattr__',
'__dict__',
'__dir__',
'__doc__',
'__eq__',
'__format__',
'__ge__',
'__getattribute__',
'__gt__',
'__hash__',
'__iadd__',
'__init__',
'__init_subclass__',
'__isub__',
'__le__',
'__lt__',
'__module__',
'__ne__',
'__neg__',
'__new__',
'__reduce__',
'__reduce_ex__',
'__repr__',
'__setattr__',
'__sizeof__',
'__str__',
'__sub__',
'__subclasshook__',
'__weakref__',
'x',
'y',
'z']

上述结果中形式为‘__函数名__’的函数为魔法函数,注意有些对象也是这种形式,例如__class__, __module__等, 这些不是魔法函数。具体的魔法函数说明可以参考Python官方说明文档。

以上代码在Python3.6运行通过.

相关文章

太好用!教你几招Python魔法方法的妙用

专注Python、AI、大数据,请关注公众号七步编程!Python是一种简单的编程语言,满足一个需求,可以有各种各样的实现方法。正是因为它可以通过各种串联满足很多复杂的逻辑,因此,对代码可读性关注度不...

11 每个程序员都应该知道的 Python 魔法方法

在 Python 中,魔法方法帮助你模拟 Python 类中内置函数的行为。这些方法有前后双下划线(__),因此也被称为魔法方法 。这些魔法方法也帮助你实现 Python 中的运算符重载。你很可能见...

Python中关于魔法方法、单例模式的知识

目录:init,del,add,str和 repr,call,单例模式,class,dict,doc,bases,mro魔法方法:定义:在特定条件下,触发方法在python里面很多以双下划线开头且结尾...

掌握Python的"魔法":特殊方法与属性完全指南

在Python的世界里,以双下划线开头和结尾的"魔法成员"(如__init__、__str__)是面向对象编程的核心。它们赋予开发者定制类行为的超能力,让自定义对象像内置类型一样优雅工...

Python 魔法方法的工作原理:实用指南(三)

以前的文章我们讲了python中对象表示、运算符等魔法方法的用法,下面我们继续深入探索容器、资源管理、性能方面的魔法方法。容器方法(类似列表、字典对象)当你需要自定义存储和检索数据的对象时,此方式可以...

Python进阶——如何正确使用魔法方法?(上)

微信搜索关注「水滴与银弹」公众号,第一时间获取优质技术干货。7年资深后端研发,用简单的方式把技术讲清楚。在做 Python 开发时,我们经常会遇到以双下划线开头和结尾的方法,例如 __init__、_...