Python:迭代器返回None

前端之家收集整理的这篇文章主要介绍了Python:迭代器返回None前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

这是我的代码

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,d):
  return list(Prizes(purchases,d))

用法示例:

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

输出应该是:

[4]

但实际输出是:

[None,None,4,None].

为什么会这样?

最佳答案
你的问题是__next__的实现.当Python调用__next__时,它总是期望返回值.但是,在您的情况下,看起来您可能并不总是每次调用都有返回值.因此,Python使用函数的默认返回值 – 无:

你需要一些方法来保持程序控制在__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
原文链接:https://www.f2er.com/python/438790.html

猜你在找的Python相关文章