python中實現可迭代對象的方法


當對元組,列表,字典,集合,字符串使用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)

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM