Python之反向迭代

需求:獲得反方向迭代一個序列
解決:使用內置的 reversed() 函數
a = [1, 2, 3, 4]
for x in reversed(a):
    print(x)   # 4 3 2 1
反向迭代僅僅當對象的大小可預先肯定或者對象實現了 __reversed__() 的特殊方法時才能生效。 若是二者都不符合,那你必須先將對象轉換爲一個列表才行
f = open('../../testData')
for line in reversed(list(f)):
    print(line, end='')
可是 要注意的是若是可迭代對象元素不少的話,將其預先轉換爲一個列表要消耗大量的內存
解決方法 經過在自定義類上實現 __reversed__() 方法來實現反向迭代
class Countdawn:
    def __init__(self,start):
        self.start=start

    def __iter__(self):
        n=self.start
        while n>0:
            yield n
            n-=1

    def __reversed__(self):
        n=1
        while n<= self.start:
            yield n
            n+=1

for r in reversed(Countdawn(5)):
    print(r)  # 1 2 3 4 5
for r in Countdawn(5):
    print(r)   # 5 4 3 2 1
相關文章
相關標籤/搜索