以下是一个示例代码,用于编写一个Python函数squareprime(l),该函数的功能是将列表l中的所有素数平方后返回一个新列表。
import math
def is_prime(n):
if n <= 1:
return False
for i in range(2, int(math.sqrt(n)) + 1):
if n % i == 0:
return False
return True
def squareprime(l):
result = []
for num in l:
if is_prime(num):
result.append(num ** 2)
return result
这段代码中,我们首先定义了一个辅助函数is_prime(n)用于判断一个数是否为素数。函数is_prime(n)通过从2到n的平方根范围内遍历,检查是否有能整除n的数,来判断n是否为素数。
然后,我们定义了主函数squareprime(l)。在这个函数里,我们创建了一个空列表result来存储结果。接着,对于列表l中的每一个元素,我们使用is_prime()函数判断其是否为素数,如果是素数,则将其平方后的结果添加到result列表中。
最后,函数返回result列表作为结果。