無法理解函數行爲 - isprime（）

Question

我已經構建了兩個函數。第（並不重要，但相關的，因爲它是所謂的第二個），講述了一個數是否是素數：

def is_prime(i): 
    if i == 1: 
     print("prime") 
    if i == 2: 
     print("not prime") 
    for d in range(2, i): 
     if i % d != 0: 
      d = d+1 
      if d == i: 
       print('prime') 
       break 
     if i % d == 0: 
      print('not prime') 
      break 

我希望能夠讓這個功能統計所有從1素數截至p。 當我要求它將它追加到列表中時，它將單獨返回一個空列表和所有值。

def prime_counting(p): 
    list_of_primes = [] 
    for n in range (p+1): 
     if is_prime(n) == "prime": 
      list_of_primes.append(n) 

我怎樣才能解決這個問題？

Answer 1

_{（注：我沒有檢查邏輯您is_prime()功能）}
在你的代碼中明顯的錯誤是函數is_prime()不return任何東西，它只是print秒。取決於數字是否爲素數，您希望它爲return True或False。改變它這樣做。

然後你可以檢查：

if is_prime(n) == True: 
    list_of_primes.append(n) 

編輯：在評論（正確地）說，寫這種說法更 「Python化」 的方法是：

if is_prime(n): 
    list_of_primes.append(n) 

由於is_prime()將返回True或False（所以沒有必要與他們比較）。

Answer 2

有代碼中的一些缺陷，但你問的是一個是：

您正在尋找的is_prime()返回值與

if is_prime(n) == "prime": 

但is_prime()不包含任何回報聲明。

這裏有一個稍微修改後的版本，實際上返回從is_prime()值：

def is_prime(i): 
    if i == 1: 
     return "prime" 
    if i == 2: 
     return "not prime" 

    for d in range(2, i): 
     if i % d != 0: 
      d = d + 1 
      if d == i: 
       return "prime" 
     if i % d == 0: 
      return "not prime" 


def prime_counting(p): 
    list_of_primes = [] 
    for n in range(p+1): 
     if is_prime(n) == "prime": 
      print("added %s" % n) 
      list_of_primes.append(n) 
    return list_of_primes 


if __name__ == '__main__': 
    print(prime_counting(3)) 

如果改爲返回一個布爾值return True或return False你將能夠只是做：

if is_prime(n): 
    do_something() 

或者甚至使用如下的「理解」：

prime_list = [n for n in range(1, p+1) if is_prime(n)]