Question

我编写了一个素数生成器程序，可以生成某个范围内的所有素数。代码如下：

from math import sqrt

primes = range(3, 31623, 2)

for i in range(len(primes)):
    if primes[i] != None:
        if primes[i] >= sqrt(31622):
            break
        for x in range(i+1, len(primes)):
            if primes[x] and not primes[x] % primes[i]:
                primes[x] = None

primes = [x for x in primes if x]

def is_prime(x):
    #if x <= 31623 and x in primes:
        #return True
    if x % 10 == 5 or not x % 10 or not x % 2:
        return False
    else:
        for i in primes:
            if not x % i or i >= sqrt(x):
                return False
    return True


l = [x for x in range(999900001, 1000000000, 2) if is_prime(x)]
#l = filter(is_prime(x), range(999900001, 1000000000, 2))
print l

问题出在最后几行。如果我使用列表理解，一切正常。但是，如果我尝试使用filter（）创建素数列表，Python将返回此错误：

Traceback (most recent call last):
  File "primes.py", line 29, in <module>
    l = filter(is_prime(x), range(999900001, 1000000000, 2))
  File "primes.py", line 20, in is_prime
    if x % 10 == 5 or not x % 10 or not x % 2:
TypeError: unsupported operand type(s) for %: 'NoneType' and 'int'

有人可以告诉我问题在哪里吗？

Answer 1

您应该将该函数传递给filter，而不是调用该函数并传递其结果。示例 -

l = filter(is_prime, range(999900001, 1000000000, 2))

问题最有可能发生，因为x被定义为None，很可能是由于列表理解 -

primes = [x for x in primes if x]

然后，当你这样做时 -

l = filter(is_prime(x), range(999900001, 1000000000, 2))

它首先评估is_prime(x)（对于x为None），这会导致问题。你真正需要做的是将函数作为第一个参数发送（不是为x评估它）。

Python filter（）不工作并返回TypeError？

1 个答案: