Python 3.7 迭代

如何解决Python 3.7 迭代

给定一个整数 n，函数 legendre_n 应该返回 n^2 和 (n+1)^2 之间的素数个数。

我的代码是这样的：

def is_prime(n):
   if n ==1:
      return False
   elif n ==2:
      return True
   elif n>2 :
      for i in range(2,n):
        if n%i ==0:
            return False
        else:
            return True
def legendre_n(n):
  counter = 0
  for i in range (n**2,(n+1)**2):
      if is_prime(i):
          counter = counter + 1
      else:
          counter+=0
   return counter
        
print(legendre_n(1))  ==2
print(legendre_n(2))  ==2
print(legendre_n(3))  ==2
print(legendre_n(12)) ==5

我应该得到上面的答案，但我得到了 2,2,4,12 知道为什么吗？

解决方法

问题在于 is_prime 方法。您在不正确的 False 循环中的 else 块中返回 for。

应该是：

def is_prime(n):
   if n == 1:
      return False
   elif n == 2:
      return True
   elif n > 2:
      for i in range(2,n):
        if n%i == 0:
            return False
      return True

,

您的 is_prime() 函数存在缺陷。

这是一个您可以使用的更简单的方法：

def isPrime(n):
    return (n==2) if n<3 else n%2 and all(n%f for f in range(3,int(n**0.5)+1,2))

     #     0,1,2     ...      odd and not divisible by any odd factor up to √n