如何定义getitem类以处理纯索引和切片?
切片对象当您使用的下标符号冒号被自动创建的-而这正是传递给__getitem__
。使用isinstance
来检查,如果你有一个切片对象:
from __future__ import print_function
class Sliceable(object):
def __getitem__(self, subscript):
if isinstance(subscript, slice):
# do your handling for a slice object:
print(subscript.start, subscript.stop, subscript.step)
else:
# Do your handling for a plain index
print(subscript)
假设我们使用的是范围对象,但我们希望切片返回列表,而不是新的范围对象(确实如此):
>>> range(1,100, 4)[::-1]
range(97, -3, -4)
由于内部限制,我们无法将范围归类,但我们可以委托给它:
class Range:
"""like builtin range, but when sliced gives a list"""
__slots__ = "_range"
def __init__(self, *args):
self._range = range(*args) # takes no keyword arguments.
def __getattr__(self, name):
return getattr(self._range, name)
def __getitem__(self, subscript):
result = self._range.__getitem__(subscript)
if isinstance(subscript, slice):
return list(result)
else:
return result
r = Range(100)
我们没有可完美替换的Range对象,但它非常接近:
>>> r[1:3]
[1, 2]
>>> r[1]
1
>>> 2 in r
True
>>> r.count(3)
1
为了更好地理解切片符号,这是Sliceable的示例用法:
>>> sliceme = Sliceable()
>>> sliceme[1]
1
>>> sliceme[2]
2
>>> sliceme[:]
None None None
>>> sliceme[1:]
1 None None
>>> sliceme[1:2]
1 2 None
>>> sliceme[1:2:3]
1 2 3
>>> sliceme[:2:3]
None 2 3
>>> sliceme[::3]
None None 3
>>> sliceme[::]
None None None
>>> sliceme[:]
None None None
Python 2,请注意:
在Python 2中,有一个不赞成使用的方法,在子类化某些内置类型时可能需要重写该方法。
从数据模型文档中:
object.__getslice__(self, i, j)
从2.0版开始不推荐使用:支持将切片对象用作__getitem__()
方法的参数。(但是,CPython中的内置类型当前仍在实现__getslice__()
。因此,在实现切片时必须在派生类中重写它。)
这在Python 3中已经消失了。
__getslice__
为python 2.X版本实现。参见docs.python.org/2/reference/datamodel.html#object.__getslice__