Python进阶之自定义对象实现切片功能


Posted in Python onJanuary 07, 2019

切片是 Python 中最迷人最强大最 Amazing 的语言特性(几乎没有之一),在《Python进阶:切片的误区与高级用法》中,我介绍了切片的基础用法、高级用法以及一些使用误区。这些内容都是基于原生的序列类型(如字符串、列表、元组......),那么,我们是否可以定义自己的序列类型并让它支持切片语法呢?更进一步,我们是否可以自定义其它对象(如字典)并让它支持切片呢?

1、魔术方法:__getitem__()

想要使自定义对象支持切片语法并不难,只需要在定义类的时候给它实现魔术方法 __getitem__() 即可。所以,这里就先介绍一下这个方法。

语法: object.__getitem__(self, key)

官方文档释义:Called to implement evaluation of self[key]. For sequence types, the accepted keys should be integers and slice objects. Note that the special interpretation of negative indexes (if the class wishes to emulate a sequence type) is up to the __getitem__() method. If key is of an inappropriate type, TypeError may be raised; if of a value outside the set of indexes for the sequence (after any special interpretation of negative values), IndexError should be raised. For mapping types, if key is missing (not in the container), KeyError should be raised.

概括翻译一下:__getitem__() 方法用于返回参数 key 所对应的值,这个 key 可以是整型数值和切片对象,并且支持负数索引;如果 key 不是以上两种类型,就会抛 TypeError;如果索引越界,会抛 IndexError ;如果定义的是映射类型,当 key 参数不是其对象的键值时,则会抛 KeyError 。

2、自定义序列实现切片功能

接下来,我们定义一个简单的 MyList ,并给它加上切片功能。(PS:仅作演示,不保证其它功能的完备性)。

class MyList():
 def __init__(self):
  self.data = []
 def append(self, item):
  self.data.append(item)
 def __getitem__(self, key):
  print("key is : " + str(key))
  return self.data[key]

l = MyList()
l.append("My")
l.append("name")
l.append("is")
l.append("Python猫")

print(l[3])
print(l[:2])
print(l['hi'])

### 输出结果:
key is : 3
Python猫
key is : slice(None, 2, None)
['My', 'name']
key is : hi
Traceback (most recent call last):
...
TypeError: list indices must be integers or slices, not str

从输出结果来看,自定义的 MyList 既支持按索引查找,也支持切片操作,这正是我们的目的。

特别需要说明的是,此例中的 __getitem__() 方法会根据不同的参数类型而实现不同的功能(取索引位值或切片值),也会妥当地处理异常,所以并不需要我们再去写繁琐的处理逻辑。网上有不少学习资料完全是在误人子弟,它们会教你区分参数的不同类型,然后写一大段代码来实现索引查找和切片语法,简直是画蛇添足。下面的就是一个代表性的错误示例:

###略去其它代码####
def __getitem__(self, index):
 cls = type(self)
 if isinstance(index, slice): # 如果index是个切片类型,则构造新实例
  return cls(self._components[index])
 elif isinstance(index, numbers.Integral): # 如果index是个数,则直接返回
  return self._components[index]
 else:
  msg = "{cls.__name__} indices must be integers"
  raise TypeError(msg.format(cls=cls))

3、自定义字典实现切片功能

切片是序列类型的特性,所以在上例中,我们不需要写切片的具体实现逻辑。但是,对于其它非序列类型的自定义对象,就得自己实现切片逻辑。以自定义字典为例(PS:仅作演示,不保证其它功能的完备性):

class MyDict():
 def __init__(self):
  self.data = {}
 def __len__(self):
  return len(self.data)
 def append(self, item):
  self.data[len(self)] = item
 def __getitem__(self, key):
  if isinstance(key, int):
   return self.data[key]
  if isinstance(key, slice):
   slicedkeys = list(self.data.keys())[key]
   return {k: self.data[k] for k in slicedkeys}
  else:
   raise TypeError

d = MyDict()
d.append("My")
d.append("name")
d.append("is")
d.append("Python猫")
print(d[2])
print(d[:2])
print(d[-4:-2])
print(d['hi'])

### 输出结果:
is
{0: 'My', 1: 'name'}
{0: 'My', 1: 'name'}
Traceback (most recent call last):
...
TypeError

上例的关键点在于将字典的键值取出,并对键值的列表做切片处理,其妙处在于,不用担心索引越界和负数索引,将字典切片转换成了字典键值的切片,最终实现目的。

4、小结

最后小结一下:本文介绍了__getitem__() 魔术方法,并用于实现自定义对象(以列表类型和字典类型为例)的切片功能,希望对你有所帮助。也希望大家多多支持三水点靠木。

Python 相关文章推荐
使用Python的Flask框架实现视频的流媒体传输
Mar 31 Python
Python的Flask框架中@app.route的用法教程
Mar 31 Python
初步介绍Python中的pydoc模块和distutils模块
Apr 13 Python
numpy.random模块用法总结
May 27 Python
Python中生成一个指定长度的随机字符串实现示例
Nov 06 Python
python使用 cx_Oracle 模块进行查询操作示例
Nov 28 Python
Jupyter 无法下载文件夹如何实现曲线救国
Apr 22 Python
详解Pandas 处理缺失值指令大全
Jul 30 Python
Python实现我的世界小游戏源代码
Mar 02 Python
OpenCV-Python实现油画效果的实例
Jun 08 Python
FP-growth算法发现频繁项集——发现频繁项集
Jun 24 Python
Python用tkinter实现自定义记事本的方法详解
Mar 31 Python
Python基于matplotlib画箱体图检验异常值操作示例【附xls数据文件下载】
Jan 07 #Python
python读取几个G的csv文件方法
Jan 07 #Python
实时获取Python的print输出流方法
Jan 07 #Python
Python 运行 shell 获取输出结果的实例
Jan 07 #Python
在python 中实现运行多条shell命令
Jan 07 #Python
Python之使用adb shell命令启动应用的方法详解
Jan 07 #Python
python 对多个csv文件分别进行处理的方法
Jan 07 #Python
You might like
php代码中使用换行及(\n或\r\n和br)的应用
2013/02/02 PHP
PHP里的单例类写法实例
2015/06/25 PHP
浅谈Yii乐观锁的使用及原理
2017/07/25 PHP
php将从数据库中获得的数据转换成json格式并输出的方法
2018/08/21 PHP
PHP常用header头定义代码示例汇总
2020/08/29 PHP
JavaScript继承方式实例
2010/10/29 Javascript
JQuery中html()方法使用不当带来的陷阱
2011/04/07 Javascript
GRID拖拽行的实例代码
2013/07/18 Javascript
jqueyr判断checkbox组的选中(示例代码)
2013/11/08 Javascript
制作jquery遮罩层效果导航菜单代码分享
2013/12/25 Javascript
页面刷新时记住滚动条的位置jquery代码
2014/06/17 Javascript
告诉你什么是javascript的回调函数
2014/09/04 Javascript
Javascript 实现图片无缝滚动
2014/12/19 Javascript
Javascript与jQuery方法的隐藏与显示
2015/01/19 Javascript
jQuery+CSS3折叠卡片式下拉列表框实现效果
2015/11/02 Javascript
浅谈jQuery animate easing的具体使用方法(推荐)
2016/06/17 Javascript
JavaScript轮播图简单制作方法
2017/02/20 Javascript
微信小程序Redux绑定实例详解
2017/06/07 Javascript
Thinkjs3新手入门之如何使用静态资源目录
2017/12/06 Javascript
利用Angular2 + Ionic3开发IOS应用实例教程
2018/01/15 Javascript
详解nodejs解压版安装和配置(带有搭建前端项目脚手架)
2018/12/06 NodeJs
微信小程序录音实现功能并上传(使用node解析接收)
2020/02/26 Javascript
Vue中 axios delete请求参数操作
2020/08/25 Javascript
使用python在校内发人人网状态(人人网看状态)
2014/02/19 Python
Python解析json文件相关知识学习
2016/03/01 Python
Django代码性能优化与Pycharm Profile使用详解
2018/08/26 Python
Python设计模式之建造者模式实例详解
2019/01/17 Python
Python应用实现处理excel数据过程解析
2020/06/19 Python
Pandas中DataFrame基本函数整理(小结)
2020/07/20 Python
意大利在线药房:Farmacia Loreto Gallo
2019/08/09 全球购物
简历里的自我评价
2014/01/31 职场文书
优秀团员事迹材料1500字
2014/08/31 职场文书
个人批评与自我批评范文
2014/10/17 职场文书
团队合作精神学习心得体会
2016/01/19 职场文书
导游词之泉州崇武古城
2019/12/20 职场文书
Flink 侧流输出源码示例解析
2022/09/23 Servers