如何為 Python 基本類型自定義方法?
我們知道,字符串str、整型int、浮點(diǎn)數(shù)float是 Python 的三大基本數(shù)據(jù)類型。
我們也知道,在 Python 里面,一切都是對(duì)象,而對(duì)象一般有方法。所以,大家會(huì)看到下面這種寫法:
- a = 'abcde'
- b = a.upper()
- print(b)
其中的.upper()就是把字母轉(zhuǎn)成大寫的方法。
我們?cè)趯懽约旱念悤r(shí),常常會(huì)定義很多的方法。那么,當(dāng)字符串自帶的方法不夠時(shí),我們是否有辦法自定義一些方法呢?
比如說(shuō),我想要一個(gè)方法叫做.are_you_short(),如果字符串的長(zhǎng)度小于5,返回 True,字符串的長(zhǎng)度大于等于5,返回 False。
我雖然可以單獨(dú)定義一個(gè)類來(lái)做這件事情:
- class MyStr:
- def __init__(self, value):
- self.value = value
- def are_you_short(self):
- return len(self.value) < 5
運(yùn)行效果如下圖所示:
但問(wèn)題是,現(xiàn)在根據(jù)MyStr類初始化的對(duì)象,它看起來(lái)完全不像一個(gè)字符串。它只是一個(gè)普通的對(duì)象了。大家對(duì)比一下普通字符串的行為和這個(gè)MyStr對(duì)象的行為:
MyStr對(duì)象,不僅打印不出內(nèi)容,而且還不能通過(guò)加號(hào)進(jìn)行拼接。
看到這里,可能有同學(xué)會(huì)認(rèn)為我要講魔術(shù)方法了。例如給MyStr實(shí)現(xiàn).__str__()方法,就能打印出內(nèi)容;實(shí)現(xiàn).__add__()方法,就可以實(shí)現(xiàn)用加號(hào)拼接。
但等等,哪里需要這么麻煩?一般有這種想法的同學(xué),是剛剛學(xué)完 Python 的魔術(shù)方法,覺(jué)得什么事情都可以用魔術(shù)方法來(lái)解決。
但實(shí)際上,要實(shí)現(xiàn)字符串的所有方法非常簡(jiǎn)單!
東西學(xué)得越多,就越容易把最簡(jiǎn)單的知識(shí)給忘記了。想想是不是還有一個(gè)東西叫做繼承?
我們修改一下MyStr的定義:
- class MyStr(str):
- def are_you_short(self):
- return len(self) < 5
運(yùn)行效果如下圖所示:
不需要手動(dòng)實(shí)現(xiàn)任何的魔術(shù)方法,直接就能擁有字符串的全部能力。
大家也可以對(duì) int、float甚至是列表、字典試一試,來(lái)實(shí)現(xiàn)自己的數(shù)據(jù)類型。
需要注意的是,在繼承 Python 的數(shù)據(jù)類型時(shí)不要手動(dòng)寫出__init__()方法,除非你知道你在干什么。