一日一技：如何为 Python 基本类型自定义方法？

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