python素数的判断（输入一个数、输入多个并求平方和）

输入单个数判断是否是素数：

def prime(n):
    flag = True
    for i in range(2, n):
        if n % i == 0 :
            flag = False
    return flag
print(prime(12))   #红色为你要判断的数

输入多个正整数判断是否是素数：

def list_prime(*n):
    list = []
    for i in n:
        flag = True
        for j in range(2, i):
            if i % j == 0:
                flag = False
                break
        if flag:
            list.append(i)
    return list
print(list_prime(7,6,8,9))   #标红为你要判断的数

输入多个正整数判断是否是素数，并且平方和：

def list_prime(*n):
    list = []
    for i in n:
        flag = True
        for j in range(2, i):
            if i % j == 0:
                flag = False
                break
        if flag:
            list.append(i)
    return list
# print(list_prime(7,6,8,9))
temp = []
prime = []
n = input("请输入你要判断的正整数，用逗号隔开：")
for i in n.split(","):
    temp.append(eval(i))
prime = list_prime(*temp)
s = 0
print("素数有：")
for j in prime:
    print(j, end=",")
for i in prime:
    s = s + i*i
print()
print("这些素数的平方和是：{}".format(s))