什么是特殊方法?當我們在設計一個類的時候,python中有一個用于初始化的方法$__init__$,類似于java中的構造器,這個就是特殊方法,也叫作魔術方法。簡單來說,特殊方法可以給你設計的類加上一些神奇的特性,比如可以進行python原生的切片操作,迭代、連乘操作等。在python中,特殊方法以雙下劃線開始,以雙下劃線結束。
一個大例子
數學中有一個表示數的概念叫做向量,但是python中的數據類型卻沒有。我們來設法用python實現它。
首先考慮,向量跟普通的數據類型不同,傳統的數可以直接進行運算,向量則需要對不同的坐標分別運算。來試試。
首先定義一個類,實現初始化方法。
# 實現向量類型 class Vector: def __init__(self, x=0, y=0): self.x = x self.y = y
如何實現向量的加法?二維向量中,向量的加法就是每個坐標分別相加得到的結果。在python中有個$__add__$方法,用來進行加法操作。
class Vector: def __init__(self, x=0, y=0): self.x = x self.y = y # 實現向量加法 def __add__(self, other): x = self.x + other.x y = self.y + other.y return Vector(x, y)
我們對x和y變量分別進行相加,然后返回Vector。在python你可以對字符串直接用加法拼接起來的原理就在此,python實現了針對字符串的add方法。
實現了加法,乘法的道理一樣,分別對每個坐標單獨相乘即可。
class Vector: def __init__(self, x=0, y=0): self.x = x self.y = y # 實現向量加法 def __add__(self, other): x = self.x + other.x y = self.y + other.y return Vector(x, y) # 實現向量乘法,例如r*3 def __mul__(self, scalar): return Vector(self.x*scalar, self.y*scalar)
我們在進行向量運算時還有一個常用的操作是求向量的模,我們用$__abs__$特殊方法來實現,abs一般用來求一個數的絕對值,向量用不到,用來求模剛好合適。使用math模塊中的hypot方法計算$\sqrt(x^2+y^2)$。
class Vector: def __init__(self, x=0, y=0): self.x = x self.y = y # 真假值,如果向量模為0,返回false def __bool__(self): return bool(abs(self)) # 實現向量加法 def __add__(self, other): x = self.x + other.x y = self.y + other.y return Vector(x, y) # 實現向量乘法,例如r*3 def __mul__(self, scalar): return Vector(self.x*scalar, self.y*scalar) # 返回向量的模 # hypot()返回歐幾里德范數 sqrt(x*x + y*y) def __abs__(self): return hypot(self.x, self.y)
找個例子運行下。
v = Vector(2, 3) print(v) v2 = Vector(4, 5) print(v+v2) print(v+v2*2)
<__main__.Vector object at 0x000002B4B1843C50> <__main__.Vector object at 0x000002B4B1843EF0> <__main__.Vector object at 0x000002B4B1843898>
可以運行了,貌似是正確的,但是輸出的結果很奇怪。怎么辦?python中有個$__repr__$特殊方法,可以修改控制臺輸出的樣式。
class Vector: def __init__(self, x=0, y=0): self.x = x self.y = y # 真假值,如果向量模為0,返回false def __bool__(self): return bool(abs(self)) # 實現向量加法 def __add__(self, other): x = self.x + other.x y = self.y + other.y return Vector(x, y) # 實現向量乘法,例如r*3 def __mul__(self, scalar): return Vector(self.x*scalar, self.y*scalar) # 返回向量的模 # hypot()返回歐幾里德范數 sqrt(x*x + y*y) def __abs__(self): return hypot(self.x, self.y) # 實現__repr__方法,在控制臺打印向量時會輸出Vector(1, 2) # 實現__str__,使用str()返回字符串 def __repr__(self): return 'Vector(%r, %r)' % (self.x, self.y)
實現了$__repr__$方法,我們就可以在控制臺輸出Vecotor(x,y)。與之對應的有個$__str__$方法,使用str()返回相應的字符串,展示給用戶。
現在來看下之前程序運行的結果。
v = Vector(2, 3) print(v) v2 = Vector(4, 5) print(v+v2) print(v+v2*2) print(abs(v))
Vector(2, 3) Vector(6, 8) Vector(10, 13) 3.605551275463989
效果不錯。
通過實現特殊方法,自定義類型可以表現的跟內置類型一樣,讓我們能夠寫出更具有python風格的代碼。
除了上面說到的幾個特殊方法外,python還有差不多80多個特殊方法,比如$__len__$方法可以用來求長度,$__getitem__$可以使用haha[2]之類的操作進行切片和迭代等,同樣的還有$__setitem__$。
更多文章、技術交流、商務合作、聯系博主
微信掃碼或搜索:z360901061

微信掃一掃加我為好友
QQ號聯系: 360901061
您的支持是博主寫作最大的動力,如果您喜歡我的文章,感覺我的文章對您有幫助,請用微信掃描下面二維碼支持博主2元、5元、10元、20元等您想捐的金額吧,狠狠點擊下面給點支持吧,站長非常感激您!手機微信長按不能支付解決辦法:請將微信支付二維碼保存到相冊,切換到微信,然后點擊微信右上角掃一掃功能,選擇支付二維碼完成支付。
【本文對您有幫助就好】元
