Python奇遇記：特殊方法窺探

什麼是特殊方法？當咱們在設計一個類的時候，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__$。

---

本人才疏學淺，上文中不免有些錯誤，還請各位品評指正。若是以爲寫的還行，歡迎關注個人公衆號MLGroup,帶你走進機器學習的世界。