交互链表时打印无而不是无

Question

我有一个链表，我在一个范围内迭代并返回所有可以表示为该范围内整数的平方数。而不是只返回可以这样做的数字，它将返回None，例如9, None, None...,16, None, None..., 25，我希望它只返回9, 16, 25 etc etc

class Squares:
    def __init__(self, start, end):
        self.__start = start - 1
        self.__end = end -1

    def __iter__(self):
        return SquareIterator(self.__start, self.__end)

class SquareIterator:
    def __init__(self, start, end): 
        self.__current = start
        self.__step = 1
        self.__end = end

    def __next__(self):
        if self.__current > self.__end:
            raise StopIteration 
        else:
            self.__current += self.__step 
        x = self.__current - self.__step + 1
        self.__current - self.__step + 1
        if str(x).isdigit() and math.sqrt(x) % 1 == 0:
            return x

Answer 1

你需要让你的__next__函数继续循环，直到达到目标值：

def __next__(self):
    # We're just going to keep looping. Loop breaking logic is below.
    while True:
        # out of bounds
        if self.__current > self.__end:
             raise StopIteration

        # We need to get the current value
        x = self.__current
        # increase the state *after* grabbing it for test
        self.__current += self.__step 

        # Test the value stored above
        if math.sqrt(x) % 1 == 0:
             return x

您应该存储 x，然后递增的原因是无论如何您都必须递增，即使您没有完美的正方形。

Answer 2

不清楚你为什么要让事情复杂化；有一个简单的方法：

import math

class Squares:
    def __init__(self, start, end):
        self.__start = start
        self.__end = end
        self.__step = 1

    def __iter__(self):
        for x in range(self.__start, self.__end, self.__step):
            if math.sqrt(x) % 1 == 0:
                yield x

s = Squares(0, 100)
for sq in s:
    print(sq, end=' ')

输出：

0 1 4 9 16 25 36 49 64 81 

来自 cmets：

请注意，避免专门的 迭代器类，只需为 Squares 实现 __iter__ 作为生成器 功能。显式__next__涉及各种低效状态 Python 做的不好的管理，而且不是那么容易遵循； __iter__ 作为生成器函数通常非常简单；每次你达到收益时，就像来自__next__ 的回报，但所有 您的状态是本地函数，不涉及特殊对象（生成器 注意保存和恢复所述本地状态）。 – 暗影游侠>

它可能甚至不需要 Squares 类。生成器函数 命名方块会做需要的事情；通过它开始，停止和步进和 将它们用作局部变量，而不是某些属性 不必要的自己。对班级唯一真正的好处是它可以 反复迭代而不重建它，可能不常见的用途 案例

def squares_from(start, stop, step=1):
    """returns a generator function for the perfect squares 
    in the range comprised between start and stop, iterated over using step=step
    """
    for x in range(start, stop, step):
        if math.sqrt(x) % 1 == 0:
            yield x

for sq in squares_from(0, 100):
    print(sq, end=' ')