摄影:产品经理产品经理给我过生日
我们知道,字符串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__()
方法,除非你知道你在干什么。