__getitem__作用
时间: 2023-10-25 20:06:52 浏览: 43
`__getitem__` 是 Python 中一个特殊的方法,用于实现对象的索引操作,即通过对象的索引来获取对象中的某个元素。
在 Python 中,可以通过 `[]` 运算符来实现索引操作,比如 `a[1]` 就是获取列表 `a` 中下标为 1 的元素。在类中定义 `__getitem__` 方法,就可以实现类似列表的索引操作,比如:
```python
class MyList:
def __init__(self, data):
self.data = data
def __getitem__(self, index):
return self.data[index]
```
上面的代码定义了一个 `MyList` 类,其中 `__getitem__` 方法实现了索引操作,可以通过 `MyList` 对象的索引来获取其中的元素,例如:
```python
lst = MyList([1, 2, 3, 4])
print(lst[0]) # 输出 1
print(lst[1]) # 输出 2
```
除了索引操作,`__getitem__` 方法还可以实现切片操作,即通过对象的切片来获取对象中的某个部分,例如:
```python
class MyList:
def __init__(self, data):
self.data = data
def __getitem__(self, index):
if isinstance(index, slice):
start, stop, step = index.indices(len(self.data))
return [self.data[i] for i in range(start, stop, step)]
else:
return self.data[index]
```
上面的代码中,如果 `index` 是一个 `slice` 对象,则说明进行的是切片操作,可以通过 `indices` 方法获取切片的起始、结束、步长三个参数,然后使用列表推导式生成切片对应的元素列表,最后返回这个列表。例如:
```python
lst = MyList([1, 2, 3, 4, 5])
print(lst[1:4]) # 输出 [2, 3, 4]
```