1. __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, 4, 5])
# 使用下标访问元素
print(my_list[0]) # 输出:1
# 使用切片操作
print(my_list[1:4]) # 输出:[2, 3, 4]
在类中定义__getitem__()
函数后,在对类进行实例化后,可实现对实例对象的索引、切片等操作,其会根据`getitem()函数中具体内容进行数值返回。
2. __iter__():
在 Python 中,__iter__
方法是一个特殊方法(也称为魔术方法或双下划线方法),用于定义一个对象的迭代行为。当你在一个类中定义了 __iter__
方法时,它会返回一个迭代器对象,该对象可以用于在 for
循环中迭代该类的实例。
下面是一个简单的示例来说明 __iter__
方法的用法:
class MyIterable:
def __init__(self, data):
self.data = data
def __iter__(self):
# 返回一个迭代器对象
return iter(self.data)
# 创建一个 MyIterable 的实例
my_iterable = MyIterable([1, 2, 3, 4, 5])
# 使用 for 循环迭代 my_iterable
for item in my_iterable:
print(item)
在这个示例中,MyIterable
类定义了一个 __iter__
方法。当你对 MyIterable
的实例使用 for
循环时,Python 首先会尝试调用该实例的 __iter__
方法来获取一个迭代器对象,然后通过调用该迭代器对象的 __next__
方法来依次获取序列中的元素。如果 __iter__
方法没有被定义,Python 将会尝试调用该实例的 __getitem__
方法来实现迭代。
通常情况下,__iter__
方法会返回一个支持迭代的对象,例如列表、元组或者自定义的迭代器对象。在上面的示例中,__iter__
方法直接返回了 self.data
的迭代器(iter()
),这样就可以在 for
循环中迭代 MyIterable
实例的 data
属性中的元素了。
如果上述代码中:
def __iter__(self):
return iter(self.data)
改为了:
def __iter__(self):
return self.data
将会报错:TypeError: iter() returned non-iterator of type 'list'
,即return
的必须是一个可迭代的对象。
当然,除了使用return
之外,还可以使用yield
来返回数据,此时,不再限制返回的值是否为可迭代对象:
class MyIterable:
def __init__(self, data):
self.data = data
def __iter__(self):
yield self.data
# 创建一个 MyIterable 的实例
my_iterable = MyIterable([1, 2, 3, 4, 5])
# 使用 for 循环迭代 my_iterable
for item in my_iterable:
print(item)
#输出:[1, 2, 3, 4, 5]