编写自定义函数，判断整形数组中的元素是否是素数

编写一个自定义函数来判断整型数组中的元素是否为素数，可以按照以下步骤：

首先，我们需要定义一个辅助函数 `is_prime` 来检查单个数字是否是素数。素数是大于1的自然数，除了1和它本身以外不再有其他正因数。

def is_prime(n):
    if n <= 1:
        return False
    elif n == 2:  # 2是最小的素数
        return True
    elif n % 2 == 0:  # 偶数不是素数，除2之外的偶数都不是
        return False
    else:  # 检查从3到√n范围内的奇数因子
        for i in range(3, int(n**0.5) + 1, 2):
            if n % i == 0:
                return False
        return True

接下来，我们可以创建一个主函数 `are_primes_in_array` ，接受一个整数数组作为输入，并对每个元素应用 `is_prime` 函数：

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

# 示例
input_array = [17, 24, 31, 46, 53]
result = are_primes_in_array(input_array)
print("数组中的素数有:", result)