當對元組,列表,字典,集合,字符串使用for循環語句的時候,可以依次拿到里面的數據,這樣的過程稱為遍歷,也叫迭代。
想要讓創建出來的類的實例對象可以迭代,也就是可以使用for來遍歷,需要在類中實現__iter__方法,需要實現__next__方法。
迭代一個可迭代對象的時候,實際上就是先獲取該對象提供的一個迭代器(__iter__),然后通過這個迭代器來依次(__next__)獲取對象中的每一個數據.
class Classmates(): def __init__(self): self.name = [] self.current_num = 0 def add(self, name): self.name.append(name) def __iter__(self): return self def __next__(self): if self.current_num < len(self.name): ret = self.name[self.current_num] self.current_num += 1 return ret else: raise StopIteration # 拋出異常停止遍歷 classmate = Classmates() classmate.add("張三") classmate.add("李四") classmate.add("王五") for name in classmate: print(name)