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

时间:2022-09-14 21:38:37

我们知道,字符串str、整型int、浮点数float是 Python 的三大基本数据类型

如何为 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 

运行效果如下图所示:

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

但问题是,现在根据MyStr类初始化的对象,它看起来完全不像一个字符串。它只是一个普通的对象了。大家对比一下普通字符串的行为和这个MyStr对象的行为:

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

MyStr对象,不仅打印不出内容,而且还不能通过加号进行拼接。

看到这里,可能有同学会认为我要讲魔术方法了。例如给MyStr实现.__str__()方法,就能打印出内容;实现.__add__()方法,就可以实现用加号拼接。

但等等,哪里需要这么麻烦?一般有这种想法的同学,是刚刚学完 Python 的魔术方法,觉得什么事情都可以用魔术方法来解决。

但实际上,要实现字符串的所有方法非常简单!

东西学得越多,就越容易把最简单的知识给忘记了。想想是不是还有一个东西叫做继承?

我们修改一下MyStr的定义:


class MyStr(str): 

    def are_you_short(self): 

        return len(self) < 5 

运行效果如下图所示:

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

不需要手动实现任何的魔术方法,直接就能拥有字符串的全部能力。

大家也可以对 int、float甚至是列表、字典试一试,来实现自己的数据类型。

需要注意的是,在继承 Python 的数据类型时不要手动写出__init__()方法,除非你知道你在干什么。