For 循环，SyntaxError：无法分配给函数调用

Question

我正在练习列表推导和嵌套列表推导。作为我练习的一部分，我正在写出等效的 for 循环。这个for循环我无法正确，我相信这是因为我试图在函数调用中分配一个值而不是一个变量。我收到的错误是：

File "<stdin>", line 4
SyntaxError: can't assign to function call

我为这个循环写的代码是：

import math

def squared_primes():
    list = []
    for x in range(1,1000000):
        for q in range(2,math.sqrt(x)+1):
            if all(x % q != 0):
                list.append(x**2)
    print(list)

这个函数试图创建一个完全平方的列表，其根是 1 到 1000000 范围内的素数。

谁能帮我理解我的循环语法到底在哪里出错了？另外，我可以将其作为嵌套列表理解吗？显然，我的列表理解正在崩溃，因为我的 for 循环语法不正确......

---

解决方案：感谢用户@Evan，我能够修复变量和语法问题，并从this thread 获取了一些关于如何修复all() 语句的提示。

此代码将正确返回 1,1000 的素数平方列表：

def squared_primes():
    list1 = []
    for x in range(1,1000):
        if all(x%q !=0 for q in range(2,int(math.sqrt(x)+1))):
            list1.append(x**2)
    print(list1)

Answer 1

此代码将正确返回从 1,1000：

除了它返回 1 作为列表的第一个元素并且 1 的平方根不是素数。让我们修复该故障并将代码重写为正确的函数：

from math import sqrt

def squared_primes(maximum):
    primes = []

    for number in range(2, maximum):
        if all(number % divisor != 0 for divisor in range(2, int(sqrt(number)) + 1)):
            primes.append(number ** 2)
    return primes

print(squared_primes(1000))

顺便说一句，这不是列表理解：

all(x % q !=0 for q in range(2, int(math.sqrt(x) + 1)))

它是一个生成器！如果你想要一个列表理解，你会这样做：

all([x % q !=0 for q in range(2, int(math.sqrt(x) + 1))])

但坚持使用生成器，因为它会以更少的努力使复合材料失败。

当我们要求提供高达 1000000（一百万）或更多的平方列表时，您的代码将开始陷入困境。那时我们需要一个更有效的基于筛的算法，例如：

def squared_primes(maximum):
    sieve = [True] * maximum

    if maximum > 0:
        sieve[0] = False  # zero is not a prime
        if maximum > 1:
            sieve[1] = False # one is not a prime

    for index in range(2, int(maximum ** 0.5) + 1):
        if sieve[index]:
            prime = index
            for multiple in range(prime + prime, maximum, prime):
                sieve[multiple] = False

    return [index * index for index in range(maximum) if sieve[index]]

大约有一百万，此代码返回结果的速度将比基于部门的解决方案快约 20 倍。

而@Evan 的光荣的理解，因为它缺乏你的math.sqrt() 优化，将比任何一个都慢几个数量级（我仍在等待它完成一百万）并开始列出两个不正确的结果。我们可以通过执行以下操作将其与您的修改后的代码相提并论：

from math import sqrt

def squared_primes(maximum):
    return [number ** 2 for number in range(2, maximum) if all(number % divisor for divisor in range(2, int(sqrt(number)) + 1))]

print(squared_primes(1000))

这是一个列表理解。但同样是错误的方法，所以回头看看基于筛子的实现。

Answer 2

这很简洁。列表推导非常棒。

def squared_primes(maximum):
    return( [ x**2 for x in range(0,maximum) if all( x % i for i in range(2, x) ) ] )

print(squared_primes(1000000))