在__getitem__中实现切片

Implementing slicing in __getitem__

提问人:nicotine 提问时间:5/30/2010 最后编辑:wjandreanicotine 更新时间:12/3/2022 访问量:102128

问:

我正在尝试为我正在创建的创建向量表示的类实现切片功能。

到目前为止,我有这段代码,我相信它会正确实现切片,但是每当我做这样的事情时,就像 where is a vector 一样,python 就会引发一个关于没有足够的参数的错误。因此,我正在尝试弄清楚如何在我的类中定义特殊方法来处理普通索引和切片。v[4]v__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]
切片 python-datamodel

评论


答:

151赞 Ignacio Vazquez-Abrams 5/30/2010 #1

当对象被切片时,该方法将接收对象。只需查看对象的 、 和 成员,即可获得切片的组件。__getitem__()slicestartstopstepslice

>>> 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')

评论

11赞 gregorySalvan 8/28/2014
注意:要扩展 list 或元组等内置类型,您必须为 python 2.X 版本实现。见 docs.python.org/2/reference/datamodel.html#object.__getslice____getslice__
0赞 Eric 8/20/2016
@gregorySalvan:该部分下面的兼容性示例不是递归吗?
3赞 user2357112 5/25/2017
@Eric:不,因为第二个冒号的存在绕过了.不过,这很微妙。__get/set/delslice__
0赞 Eric 5/25/2017
@user2357112:哇,完全错过了第二个冒号——谢谢!
0赞 wjandrea 4/10/2019
@alancalvitti IIRC 中,这是为了在 Python 2 中创建新样式的类。
8赞 carl 5/30/2010 #2

执行此操作的正确方法是__getitem__取一个参数,该参数可以是数字或切片对象。

87赞 3 revs, 2 users 78%Walter Nissen #3

我有一个“合成”列表(其中数据大于您想要在内存中创建的列表),我的列表如下所示:__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."

切片不返回相同的类型,这是禁忌,但它对我有用。

评论

1赞 estan 6/2/2016
如果键 >= len( self ) 不应该是 if 键 < 0 或键 >= len( self ) 吗?如果传递了密钥 < -len(self) 怎么办?
35赞 Russia Must Remove Putin 12/19/2015 #4

如何定义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 对象,但我们希望切片返回列表而不是新的 range 对象(因为它确实如此):

>>> 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 版起不推荐使用:支持切片对象作为方法的参数。(但是,CPython 中的内置类型目前仍实现 .因此,在实现切片时,必须在派生类中重写它。__getitem__()__getslice__()

这在 Python 3 中消失了。

评论

0赞 Jitin 1/13/2022
当我们处理普通索引时,我们不能调用 self[index],因为它会进入递归,你如何访问正确的元素?
1赞 Russia Must Remove Putin 1/14/2022
如果要使用父级对已有方法的实现,请使用 。看到 stackoverflow.com/questions/222877/...super()
9赞 Eric Cousineau 5/25/2017 #5

为了扩展 Aaron 的答案,对于类似 的东西,你可以通过检查是否是 来进行多维切片:numpygiventuple

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)))

评论

0赞 Eric Cousineau 5/30/2017
作为次要的后续操作,下面是一个使用它来映射 MATLAB 索引和 NumPy 索引(目前 MATLAB R2016b 不支持)的示例,并提供了它的示例用法。