以下是一个示例代码,用于编写一个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
列表作为结果。