📅  最后修改于: 2023-12-03 14:58:04.821000             🧑  作者: Mango
在编写程序时,有时需要将数组中的元素进行修改,使其只保留素数。以下是一些方法,可以通过合并数组或添加新的素数元素来实现这一操作。
可以创建一个空的数组,然后遍历原数组中的每个元素,检查其是否为素数,并将其添加到新数组中。
# 定义一个函数来检查素数
def check_prime(n):
if n < 2:
return False
for i in range(2, int(n ** 0.5) + 1):
if n % i == 0:
return False
return True
# 创建一个空的数组
new_array = []
# 遍历原数组中的所有元素
for element in original_array:
# 如果元素是素数,则将其添加到新数组中
if check_prime(element):
new_array.append(element)
# new_array 现在是只包含素数的数组
如果原数组中只包含少量元素,可以直接添加新的素数元素,而无需创建一个新的数组。
# 定义一个函数来检查素数
def check_prime(n):
if n < 2:
return False
for i in range(2, int(n ** 0.5) + 1):
if n % i == 0:
return False
return True
# 对原数组中的每个元素进行检查
for i in range(len(original_array)):
# 如果元素不是素数,则替换为下一个素数
if not check_prime(original_array[i]):
for j in range(original_array[i] + 1, original_array[i] * 2):
if check_prime(j):
original_array[i] = j
break
# original_array 现在是只包含素数的数组
以上是两种不同的方法,可以根据实际情况选择其中一种适用的方法来处理数组。