slice - Python:在__getitem__中实现切片
我正在尝试为我正在创建的类创建切片功能,以创建矢量表示。
到目前为止我有这个代码,我相信这将正确实现切片,但每当我做一个像v[4]这样的调用,其中v是一个向量python返回一个关于没有足够的参数的错误。 所以我试图弄清楚如何在我的类中定义getitem特殊方法来处理普通索引和切片。
def __getitem__(self, start, stop, step):
index = start
if stop == None:
end = start + 1
else:
end = stop
if step == None:
stride = 1
else:
stride = step
return self.__data[index:end:stride]
5个解决方案
102 votes
切片对象时,__getitem__()方法将收到slice对象。 只需查看start,stop和step成员的slice对象,即可获得切片的组件。
>>> class C(object):
... def __getitem__(self, val):
... print val
...
>>> c = C()
>>> c[3]
3
>>> c[3:4]
slice(3, 4, None)
>>> c[3:4:-2]
slice(3, 4, -2)
>>> c[():1j:'a']
slice((), 1j, 'a')
Ignacio Vazquez-Abrams answered 2019-08-22T22:40:29Z
59 votes
我有一个"合成" list(数据大于你想要在内存中创建的那个)和我的__getitem__看起来像这样:
def __getitem__( self, key ) :
if isinstance( key, slice ) :
#Get the start, stop, and step from the slice
return [self[ii] for ii in xrange(*key.indices(len(self)))]
elif isinstance( key, int ) :
if key < 0 : #Handle negative indices
key += len( self )
if key < 0 or key >= len( self ) :
raise IndexError, "The index (%d) is out of range."%key
return self.getData(key) #Get the data from elsewhere
else:
raise TypeError, "Invalid argument type."
切片不会返回相同的类型,这是禁忌,但它适用于我。
Walter Nissen answered 2019-08-22T22:41:02Z
10 votes
如何定义getitem类来处理普通索引和切片?
在下标表示法中使用冒号时会自动创建切片对象 - 这就是传递给__getitem__()的内容。使用__getslice__()检查是否有切片对象:
from __future__ import print_function
class Sliceable(object):
def __getitem__(self, given):
if isinstance(given, slice):
# do your handling for a slice object:
print(given.start, given.stop, given.step)
else:
# Do your handling for a plain index
print(given)
用法示例:
>>> 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中,有一个不推荐使用的方法,在子类化一些内置类型时可能需要覆盖它。
从datamodel文档:
__getitem__()
自2.0版本后不推荐使用:支持切片对象作为__getitem__()方法的参数。 (但是,CPython中的内置类型目前仍然实现__getslice__().因此,在实现切片时必须在派生类中重写它。)
这在Python 3中消失了。
Aaron Hall answered 2019-08-22T22:42:19Z
7 votes
正确的方法是让__getitem__获取一个参数,该参数可以是数字,也可以是切片对象。
看到:
[http://docs.python.org/library/functions.html#slice]
[http://docs.python.org/reference/datamodel.html#object.__getitem__]
carl answered 2019-08-22T22:43:02Z
6 votes
要扩展Aaron的答案,对于像numpy这样的事情,您可以通过检查given是否为tuple来进行多维切片:
class Sliceable(object):
def __getitem__(self, given):
if isinstance(given, slice):
# do your handling for a slice object:
print("slice", given.start, given.stop, given.step)
elif isinstance(given, tuple):
print("multidim", given)
else:
# Do your handling for a plain index
print("plain", given)
sliceme = Sliceable()
sliceme[1]
sliceme[::]
sliceme[1:, ::2]
```
输出:
('plain', 1)
('slice', None, None, None)
('multidim', (slice(1, None, None), slice(None, None, 2)))
Eric Cousineau answered 2019-08-22T22:43:34Z