这个问题要求我编写一个函数,它从给定函数的输出中编译一个素数列表,该函数给出2个列表,一个从2到任意数字的数字列表,以及与第一个列表中的数字匹配的第二个列表。 “true”或“False”值取决于第一个列表中的数字是否为素数。
我不知道我的代码是否从根本上错误地回答了这个问题,或者我是否在正确的轨道上并且只是犯了错误...
任何帮助都会非常感激。
问题:
编写一个带有单个输入N的函数(称为primes_list)。该函数必须使用prime_sieve函数来计算并返回仅小于或等于N + 1的素数的数组(或列表)。
例如,如果N = 8,则此函数应返回[2,3,5,7]
给定的代码:
import numpy as np
def prime_sieve(N):
nums = np.arange(2, N+2, 1)
mask = []
for n in nums:
mask.append(True)
for n in nums:
for i in np.arange(2*n-2, N, n):
mask[i] = False
return nums, np.array(mask)
numbers, mask = prime_sieve(8)
print(numbers)
print(mask)
[2 3 4 5 6 7 8 9]
[ True True False True False True False False]
我的代码:
import numpy as np
def primes_list(N):
numbers, mask = prime_sieve(N)
primes = []
for n in numbers:
if mask[n] == "true":
primes.append(numbers[n])
return primes
print(primes_list(8))
但这会给出一个错误:
---------------------------------------------------------------------------
IndexError Traceback (most recent call last)
<ipython-input-60-4ea4d2f36734> in <module>
----> 2 print(primes_list(8))
<ipython-input-59-a5080837c5c8> in primes_list(N)
6 primes = []
7 for n in numbers:
----> 8 if mask[n] == "true":
9 primes.append(numbers[n])
10 return primes
IndexError: index 8 is out of bounds for axis 0 with size 8
您用来切割列表n
的mask
是一个不适合索引的数字列表(因为它总是包含N,N + 1,而mask
的最后一个索引是N-1)。
此外,第二个列表mask
包含Bool
而不是str
,所以你对mask[n] == 'true'
的比较将总是返回False
。
考虑到以上几点,您的primes_list
可以是:
def primes_list(N):
numbers, mask = prime_sieve(N)
primes = []
for i, n in enumerate(numbers): # <<< added enumerate
if mask[i]: # <<< removed unnecessary comparison
primes.append(n) # <<< append n directly
return primes
返回:
[2, 3, 5, 7]
应该如此。