__getitem__方法
时间: 2023-10-25 22:08:12 浏览: 36
在 Python 中,__getitem__ 方法是一种特殊方法,用于让对象像序列一样支持索引操作。当我们使用索引操作符 [] 来获取一个对象的元素时,Python 会自动调用该对象的 __getitem__ 方法,并将索引作为参数传递给该方法。__getitem__ 方法需要返回与索引对应的元素。
例如,我们可以定义一个自定义列表类,实现 __getitem__ 方法,使其支持索引操作:
```
class MyList:
def __init__(self, data):
self.data = data
def __getitem__(self, index):
return self.data[index]
```
然后我们可以创建一个 MyList 对象并像使用列表一样进行索引操作:
```
my_list = MyList([1, 2, 3])
print(my_list[0]) # 输出 1
print(my_list[1]) # 输出 2
print(my_list[2]) # 输出 3
```
除了支持索引操作外,__getitem__ 方法还可以支持切片操作。当我们使用切片操作符 [start:end:step] 时,Python 会自动调用该对象的 __getitem__ 方法,并将切片对象作为参数传递给该方法。__getitem__ 方法需要返回与切片对象对应的元素序列。
例如,我们可以在上面的 MyList 类中添加对切片操作的支持:
```
class MyList:
def __init__(self, data):
self.data = data
def __getitem__(self, index_or_slice):
if isinstance(index_or_slice, int):
return self.data[index_or_slice]
else:
start, stop, step = index_or_slice.indices(len(self.data))
return [self.data[i] for i in range(start, stop, step)]
```
然后我们可以像使用列表一样进行切片操作:
```
my_list = MyList([1, 2, 3, 4, 5])
print(my_list[1:4]) # 输出 [2, 3, 4]
print(my_list[::2]) # 输出 [1, 3, 5]
```
阅读全文