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

作者:admin 发布日期:2020年12月22日 05:46

我们知道,字符串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

运行效果如下图所示:

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

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

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

 

原文链接:https://mp.weixin.qq.com/s/1PKrDNd8HURnplBUryQZoQ

文章标签: python
本篇共 0 条评论

留言内容:

还没有任何评论!

还没有登陆,请登录后发表评论!
文章目录