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 - 循环：

来完成

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。

Python：迭代器返回None

2 个答案: