소수 판별 함수 만들기 [파이썬 기초]

소수는 1과 자신 이외의 자연수로는 나눌 수 없는 수를 말한다. 어떤 숫자를 입력 =하면 그 수가 소수인지 판별하는 함수를 만들어보자

 

 

소수 판별 함수

 

[파이썬 소스 코드]

def is_prime(num):
    if num == 2:
        return True
    for i in range(2, num):
        if num % i == 0 :
            return False
    return True

while True:
    try:
        number = int(input('\n숫자를 입력하세요.(종료는 "0") : '))
    except:
        continue

    if not number:
        break
    else:
        result = is_prime(number)
        if result:
            print(f'{number} : 소수')
        else :
            print(f'{number} : 소수 아님')

 

[실행 결과]

숫자를 입력하세요.(종료는 "0") : 2
2 : 소수

숫자를 입력하세요.(종료는 "0") : 9
9 : 소수 아님

숫자를 입력하세요.(종료는 "0") : 18
18 : 소수 아님

숫자를 입력하세요.(종료는 "0") : 3321
3321 : 소수 아님

숫자를 입력하세요.(종료는 "0") : 371
371 : 소수 아님

숫자를 입력하세요.(종료는 "0") : 121
121 : 소수 아님

숫자를 입력하세요.(종료는 "0") : 167
167 : 소수

숫자를 입력하세요.(종료는 "0") : 0