python 全WA

from math import*
def func(x):
    flag = 1
    if x <= 2:
        flag = 1
    else:
        for i in range(2, int(x / 2 + 1)):
            if x % i == 0:
                flag = 0
                break
    return flag
def main():
    a = input()
    b = list(map(int, input().split()))
    for x in b:
        if func(x) == 0:
            b.remove(x)
    for i in b:
        print(i, end=" ")
main()