การใช้การแบ่งส่วนใน __getitem__


112

ฉันกำลังพยายามใช้ฟังก์ชัน slice สำหรับคลาสที่ฉันกำลังสร้างซึ่งสร้างการแสดงเวกเตอร์

จนถึงตอนนี้ฉันมีรหัสนี้ซึ่งฉันเชื่อว่าจะใช้ชิ้นส่วนได้อย่างถูกต้อง แต่เมื่อใดก็ตามที่ฉันทำการโทรเช่น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]

คำตอบ:


118

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

10
หมายเหตุ: สำหรับการขยายประเภทบิวด์อินเช่นรายการหรือทูเพิลคุณต้องใช้__getslice__สำหรับเวอร์ชัน python 2.X ดูdocs.python.org/2/reference/datamodel.html#object.__getslice__
gregorySalvan

@gregorySalvan: ตัวอย่างความเข้ากันได้ด้านล่างส่วนนั้นไม่ได้ถูกเรียกคืนใช่หรือไม่?
Eric

3
@ เอริก: ไม่เพราะการมีลำไส้ใหญ่ที่สองข้าม__get/set/delslice__ไป มันค่อนข้างบอบบาง
user2357112 รองรับ Monica

@ user2357112: ว้าวพลาดโคลอนที่สองไปโดยสิ้นเชิงขอบคุณ!
Eric

@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
ไม่ควร if key> = len (self) be if key <0 or key> = len (self)? จะเกิดอะไรขึ้นถ้าคีย์ <-len (self) ถูกส่งผ่าน?
estan

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

เพื่อให้เข้าใจสัญกรณ์ slice ได้ดีขึ้นตัวอย่างการใช้ 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

หากต้องการขยายคำตอบของ 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)))

ในฐานะที่เป็นผู้เยาว์ติดตามที่นี่เป็นตัวอย่างของการจ้างนี้เพื่อ map ระหว่างการจัดทำดัชนี MATLAB และ NumPy การจัดทำดัชนี (ซึ่งปัจจุบันไม่ได้รับการสนับสนุนใน MATLAB R2016b) โดยมีตัวอย่างการใช้งานของมัน
Eric Cousineau

โดยการใช้ไซต์ของเรา หมายความว่าคุณได้อ่านและทำความเข้าใจนโยบายคุกกี้และนโยบายความเป็นส่วนตัวของเราแล้ว
Licensed under cc by-sa 3.0 with attribution required.