def p(i):

    if i==1 or (i!=2 and i%2==0):

        return False

    else:

        for j in range(2, int(i**0.5)+1):

            if i%j==0:

               return False

        return True


이걸로 몇 개를 푼 건지 ㅋㅋㅋ