python - 具有 2 个迭代器和 2 个范围的 for 循环

Question

我是 Python 新手，正在处理素数生成器。我想使用的算法是 Sieve of Atkin .

此刻，我正在尝试按照算法的伪代码作为我的练习。但是，我遇到了一个问题，我找不到任何关于它的引用。 (也许我不擅长搜索......)。在伪代码中，for (x, y) in [1, √limit] × [1, √limit]:... 让我感到困惑。我知道这是什么意思，但不知道如何将这段代码翻译成 Python 代码。

如果我的问题不合适，请见谅，感谢您的帮助。 :)

下面是我的部分代码。

itx = iter(range(1, int(limit**0.5)))
ity = iter(range(1, int(limit**0.5)))
for x, y in zip(itx, ity):
    n = 4*(x**2)+(y**2)
    if n <= limit and (n%12 == 1 or n%12 == 5):
        sieve[n] = not sieve[n]

    n = 3*(x**2)+(y**2)
    if n <= limit and n%12 == 7:
        sieve[n] = not sieve[n]

    n = 3*(x**2)-(y**2)
    if x > y and n <= limit and n%12 == 11:
        sieve[n] = not sieve[n]

    itx.next()
    ity.next()

Answer 1

for (x, y) in [1, √limit] × [1, √limit] 应该翻译成乘积:

for x in itx:
    for y in ity:

或使用 itertools.product() :

from itertools import product

for x, y in product(itx, ity):

请注意，您不需要需要在迭代器上调用.next()!删除 itx.next() 和 ity.next() 行，除非您意思跳过生成的值。 for 构造为您改进迭代器。