我们知道,字符串str、整型int、浮点数float是 Python 的三大基本数据类型。
我们也知道,在 Python 里面,一切都是对象,而对象一般有方法。所以,大家会看到下面这种写法:
a = 'abcde'
b = a.upper()
print(b)
其中的.upper()就是把字母转成大写的方法。
我们在写自己的类时,常常会定义很多的方法。那么,当字符串自带的方法不够时,我们是否有办法自定义一些方法呢?
比如说,我想要一个方法叫做.are_you_short(),如果字符串的长度小于5,返回 True,字符串的长度大于等于5,返回 False。
我虽然可以单独定义一个类来做这件事情:
class MyStr:
def __init__(self, value):
self.value = value
def are_you_short(self):
return len(self.value) < 5
运行效果如下图所示:
但问题是,现在根据MyStr类初始化的对象,它看起来完全不像一个字符串。它只是一个普通的对象了。大家对比一下普通字符串的行为和这个MyStr对象的行为:
MyStr对象,不仅打印不出内容,而且还不能通过加号进行拼接。
看到这里,可能有同学会认为我要讲魔术方法了。例如给MyStr实现.__str__()方法,就能打印出内容;实现.__add__()方法,就可以实现用加号拼接。
但等等,哪里需要这么麻烦?一般有这种想法的同学,是刚刚学完 Python 的魔术方法,觉得什么事情都可以用魔术方法来解决。
但实际上,要实现字符串的所有方法非常简单!
东西学得越多,就越容易把最简单的知识给忘记了。想想是不是还有一个东西叫做继承?
我们修改一下MyStr的定义:
class MyStr(str):
def are_you_short(self):
return len(self) < 5
运行效果如下图所示:
不需要手动实现任何的魔术方法,直接就能拥有字符串的全部能力。
大家也可以对 int、float甚至是列表、字典试一试,来实现自己的数据类型。
需要注意的是,在继承 Python 的数据类型时不要手动写出__init__()方法,除非你知道你在干什么。