python 匿名函數實現求素數平方和

sq=lambda x:x**2
def list_prime(*data):
    list1=[]
    list2=[]
    for i in data:
        list1.append(i)
    for i in range(len(list1)):
        a=0
        for j in range(2,list1[i]):
            if list1[i]%j==0:
                a=1
                break
        if a==0 and list1[i]>1:
            list2.append(list1[i])
    return list2
c=list(map(int, input("請輸入一串數字（用空格分隔）：").split()))
print("素數有：{}".format(list_prime(*c)))
list3=[]
for i in list_prime(*c):
    list3.append(i)
print("素數的平方和分別為：{}".format(list3))