__getitem__でのスライスの実装


112

ベクター表現を作成する、作成しているクラスのスライス機能を実装しようとしています。

これまでのところ、このコードはありますが、スライスは適切に実装されると思いますが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]

回答:


118

__getitem__()この方法は、受信するsliceオブジェクトをスライスするときにオブジェクトを。単純に見てstartstopstepのメンバー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')

10
注:リストやタプルなどの組み込み型を拡張するには__getslice__、Python 2.Xバージョンに実装する必要があります。docs.python.org/2/reference/datamodel.html#object.__getslice__
gregorySalvanを

@gregorySalvan:そのセクションの下の互換性の例はただ再帰していませんか?
Eric

3
@エリック:いいえ、2番目のコロンの存在はバイパスし__get/set/delslice__ます。ただし、かなり微妙です。
user2357112は2017

@ user2357112:うわー、2番目のコロンを完全に逃した-ありがとう!
エリック

@alancalvitti IIRC、Pythonの2の新しいスタイルのクラスを作成するためだ
wjandrea

64

私は「合成」リスト(データがメモリに作成するよりも大きいリスト)を持っていますが、__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
key> = len(self)の場合、key <0またはkey> = len(self)の場合はどうでしょうか?キー<-len(self)が渡された場合はどうなりますか?
2016年

20

プレーンインデックスとスライスの両方を処理する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には、いくつかの組み込み型をサブクラス化するときにオーバーライドする必要がある非推奨のメソッドがあります。

datamodelドキュメントから:

object.__getslice__(self, i, j)

バージョン2.0で非推奨:__getitem__()メソッドのパラメーターとしてスライスオブジェクトをサポートします。(ただし、CPythonの組み込み型は現在も実装してい__getslice__()ます。そのため、スライスを実装するときに、派生クラスでそれをオーバーライドする必要があります。)

これはPython 3ではなくなりました。


7

アーロンの答えを拡張するために、などの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)))

マイナーのフォローアップとして、ここでは一例であると、MATLABのインデックスと(現在MATLAB R2016bではサポートされていません)numpyのインデックス作成の間でマッピングするために、これを用いた使用例、それの。
Eric Cousineau

弊社のサイトを使用することにより、あなたは弊社のクッキーポリシーおよびプライバシーポリシーを読み、理解したものとみなされます。
Licensed under cc by-sa 3.0 with attribution required.