Python：迭代器返回None

Question

這是我的代碼：

class Prizes(object):
    def __init__(self, purchases, n, d):
        self.p = purchases
        self.n = n
        self.d = d
        self.x = 1

    def __iter__(self):
        return self

    def __next__(self):
        print(self.x)

        if self.x % self.n == 0 and self.p[self.x - 1] % self.d == 0:
            self.x = self.x + 1
            return self.x - 1
        elif self.x > len(self.p):
            raise StopIteration

        self.x = self.x + 1

def superPrize(purchases, n, d):
  return list(Prizes(purchases, n, d))

用法示例：

superPrize([12, 43, 13, 465, 1, 13], 2, 3)

輸出應該是：

[4]

但實際輸出是：

[None, None, None, 4, None, None].

為什么會這樣？

Answer 1

你的問題是__next__的實現。 當Python調用__next__ ， 它總是期望返回值 。 但是，在您的情況下，看起來您可能並不總是每次調用都有返回值。 因此，Python使用函數的默認返回值 - None ：

你需要一些方法來保持程序控制在__next__直到你有一個實際的返回值。 這可以使用while -loop來完成：

def __next__(self):
    while True:
        if self.x % self.n == 0 and self.p[self.x - 1] % self.d == 0:
            self.x = self.x + 1
            return self.x - 1
        elif self.x > len(self.p):
            raise StopIteration
        self.x = self.x + 1

Answer 2

用while包裝它，這樣你的方法就不會返回值，直到找到一個：

def __next__(self):
    while True:
        if self.x % self.n == 0 and self.p[self.x - 1] % self.d == 0:
            self.x = self.x + 1
            return self.x - 1
        elif self.x > len(self.p):
            raise StopIteration

        self.x = self.x + 1

使用迭代器的事情調用__next__期望它返回一個值，但該方法僅在條件下返回一個值，否則它到達方法的末尾並返回None 。