python中的定製類(形如__xxx__的變量)很有意思,發現很重要(持續更新)

剛開始研究這個,是在看pytorch框架內的具體類的源代碼的時候,發現這個很重要,來安利一波

 

python中除了可以使用內建的類型,如list,tuple,dict,還可以創建自己的對象來實現像這些內建類型的訪問,不過需要在定義類的時候對一些魔法方法逐一實現。

1、__str__

我們先定義一個Study類,打印一個實例:

打印出一堆<__main__>,不好看,怎麼才能打印出具體的值呢?只需要定義好__str__()方法,返回一個好看的字符串就可以了

這是因爲直接顯示變量調用的不是__str__(),而是__repr__(),兩者的區別是__str__()返回用戶看到的字符串,而__repr__()返回程序開發者看到的字符串,也就是說,__repr__()是爲調試服務的。

解決辦法是再定義一個__repr__()。但是通常__str__()__repr__()代碼都是一樣的,所以,有個偷懶的寫法:

__repr__=__str__

此例表明:同時定義 __repr__ 方法和 __str__ 方法時,print() 方法會調用 __str__ 方法。

__repr__方法是在直接輸入的時候,會顯示,具體參考下面的鏈接

參考:https://blog.csdn.net/sinat_41104353/article/details/79254149

2、__iter__

如果一個類想被用於for ... in循環,類似list或tuple那樣,就必須實現一個__iter__()方法,該方法返回一個迭代對象,然後,Python的for循環就會不斷調用該迭代對象的__next__()方法拿到循環的下一個值,直到遇到StopIteration錯誤時退出循環。

具體還可以參考這篇博客:https://blog.csdn.net/zuyuhuo6777/article/details/85857311

我們以斐波那契數列爲例,寫一個Fib類,可以作用於for循環:

class Fib(object):
    def __init__(self):
        self.a, self.b = 0, 1 # 初始化兩個計數器a,b

    def __iter__(self):
        return self # 實例本身就是迭代對象,故返回自己

    def __next__(self):
        self.a, self.b = self.b, self.a + self.b # 計算下一個值
        if self.a > 20: # 退出循環的條件
            raise StopIteration()
        return self.a # 返回下一個值

現在,我們來吧Fib實例用於For循環:

3、__getitem__

現在好像和list有點像,但是,把它當做list還是不行,比如:我現在去第一個元素:

要表現的像list那樣按照下標取出元素,需要實現__getitem__()方法:

    def __getitem__(self,n):
        a,b=1,1
        for x in range(n):
            a,b=b,a+b
        return a

現在,就可以按下標訪問數列的任意一項:

但是list有個神奇的切片方法:

對於Fib卻報錯。原因是__getitem__()傳入的參數可能是一個int,也可能是一個切片對象slice,所以要做判斷:

class Fib(object):
    def __getitem__(self, n):
        if isinstance(n, int): # n是索引
            a, b = 1, 1
            for x in range(n):
                a, b = b, a + b
            return a
        if isinstance(n, slice): # n是切片
            start = n.start
            stop = n.stop
            if start is None:
                start = 0
            a, b = 1, 1
            L = []
            for x in range(stop):
                if x >= start:
                    L.append(a)
                a, b = b, a + b
            return L

現在試試Fib的切片

>>> f = Fib()
>>> f[0:5]
[1, 1, 2, 3, 5]
>>> f[:10]
[1, 1, 2, 3, 5, 8, 13, 21, 34, 55]

但是沒有對step參數作處理:

>>> f[:10:2]
[1, 1, 2, 3, 5, 8, 13, 21, 34, 55, 89]

也沒有對負數作處理,所以,要正確實現一個__getitem__()還是有很多工作要做的。

更加詳細的內容可以參考:https://www.liaoxuefeng.com/wiki/0014316089557264a6b348958f449949df42a6d3a2e542c000/0014319098638265527beb24f7840aa97de564ccc7f20f6000

總之,通過上面的方法,我們自己定義的類表現得和Python自帶的list、tuple、dict沒什麼區別,這完全歸功於動態語言的“鴨子類型”,不需要強制繼承某個接口。

4、__getattr__

__getattr__爲內置方法,當使用點號獲取實例屬性時,如果屬性不存在就自動調用__getattr__方法

5、__setattr__

__setattr__當設置類實例屬性時自動調用,如j.name=5 就會調用__setattr__方法  self.[name]=5

6、__call__

爲了將一個類實例當做函數調用,我們需要在類中實現__call__()方法。

https://blog.csdn.net/Yaokai_AssultMaster/article/details/70256621

 

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章