[python] 효율적인 소수 판별

import math

def isPrime2(n):
    # To-do
    # [실습2]의 코드를 더 효율적으로 발전시켜보세요!
    if n == 2 or n == 3:
        return True
    if n % 2 == 0 or n == 1:
        return False
    for i in range(3, int(math.sqrt(n))+1, 2):
        if n % i == 0:
            return False
    return True
    

# 결과 출력을 위한 코드입니다. 자유롭게 값을 바꿔보며 확인해보세요.
print(isPrime2(1))