将一个数组中的素数生成新的数组,并输出新数组python

在Python中，我们可以编写一个函数来遍历给定的数组，检查每个元素是否为素数，然后将素数添加到一个新的数组中。以下是简单的实现步骤：

def is_prime(n):
    if n <= 1:
        return False
    for i in range(2, int(n**0.5) + 1):
        if n % i == 0:
            return False
    return True

def primes_from_array(arr):
    prime_arr = []
    for num in arr:
        if is_prime(num):
            prime_arr.append(num)
    return prime_arr

# 示例
input_array = [1, 2, 3, 4, 5, 6, 7, 8, 9]
new_array = primes_from_array(input_array)
print("新数组只包含素数:", new_array)