如何为 Python 基本类型自定义方法？

上一篇 / 下一篇 2020-10-20 10:17:36

　　我们知道，字符串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__()方法，除非你知道你在干什么。