web-dev-qa-db-ja.com

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]
96
nicotine

__getitem__()メソッドは、オブジェクトがスライスされるときにsliceオブジェクトを受け取ります。スライスのコンポーネントを取得するために、startオブジェクトのstopstep、および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')
107

「合成」リスト(メモリに作成したいデータよりも大きいリスト)と__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."

スライスは同じタイプを返しません。これはノーです。しかし、私にとってはうまくいきます。

61
Walter Nissen

Getitemクラスを定義して、プレーンインデックスとスライスの両方を処理する方法は?

添え字表記でコロンを使用すると、スライスオブジェクトが自動的に作成されます-thatは___getitem___に渡されます。 isinstanceを使用して、スライスオブジェクトがあるかどうかを確認します。

_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 documentation から:

object.__getslice__(self, i, j)

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

これはPython 3。

13
Aaron Hall

これを行う正しい方法は、__getitem__は、1つのパラメーターを取ります。パラメーターは、数値またはスライスオブジェクトのいずれかです。

見る:

http://docs.python.org/library/functions.html#slice

http://docs.python.org/reference/datamodel.html#object.__getitem__

7
carl

アーロンの答えを拡張するために、numpyのようなものについて、givenTupleであるかどうかを確認することにより、多次元スライスを行うことができます。

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)))
6
Eric Cousineau