입력한 수만큼 소수 추출하기 [파이썬 기초 예제]

소수의 개수를 구하는 문제인데, 입력한 수만큼 소수를 추출하여 출력한다. 프로그램을 처음 시작할 때 소수 구하는 알고리즘을 통해 반복문과 조건문을 많이 연습하는데, 이 코드는 함수를 만들어서 이 부분을 처리한다.

 

 

원하는 개수대로 소수 출력

 

[파이썬 소스 코드]

def is_prime(n):
    prime = True
    if n < 2:
        prime = False
    else:
        for i in range(2, n):
            if n % i == 0:
                prime = False
                break
    return prime


def get_prime(n):
    count = 0
    result = []
    number = 2

    while True:
        res = is_prime(number)
        if res:
            result.append(number)
            count += 1
        number += 1
        if count == n:
            break
    return result


if __name__ == '__main__':
    while True:
        try:
            num = int(input('\n추출할 소수 개수 입력(종료:"0")? '))
        except:
            print('숫자만 입력!')
            continue
        
        if not num:
            break
        else:
            result = get_prime(num)
            print(result)

• get_prime() 함수의 인자 n에 입력받은 num을 대입하면 그 숫자만큼의 소수를 추출하기 위해 2부터 숫자 1씩 증가시키면서 그 수를 is_prime() 함수의 인자 n에 대입하여 호출한다.
• is_prime() 함수는 소수 여부를 판단한 후 리턴한다.
• 다시 get_prime() 함수는 is_prime() 함수의 리턴값이 소수(True)이면 result 리스트에 추가하고 count를 1 증가시킨다.
• 마지막으로 입력한 소수 추출 숫자(num)와 count가 같으면 반복문 while을 종료하고 결과(result 리스트)를 리턴하고,
  결과를 받은 후 이를 출력한다.

 

 

[실행 결과]

추출할 소수 개수 입력(종료:"0")? 4
[2, 3, 5, 7]

추출할 소수 개수 입력(종료:"0")? 10
[2, 3, 5, 7, 11, 13, 17, 19, 23, 29]

추출할 소수 개수 입력(종료:"0")? 20 
[2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37, 41, 
43, 47, 53, 59, 61, 67, 71]

추출할 소수 개수 입력(종료:"0")? 100
[2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37, 41, 
43, 47, 53, 59, 61, 67, 71, 73, 79, 83, 89, 97, 101, 103, 107, 109, 113, 127, 131, 137, 139, 149, 
151, 157, 163, 167, 173, 179, 181, 191, 193, 197, 199, 211, 223, 227, 229, 233, 239, 241, 251, 257, 263, 269, 271, 277, 281, 283, 293, 307, 311, 313, 317, 331, 337, 347, 349, 353, 359, 367, 373, 379, 383, 389, 397, 401, 409, 419, 421, 431, 433, 
439, 443, 449, 457, 461, 463, 467, 479, 487, 491, 499, 503, 509, 521, 523, 541]

추출할 소수 개수 입력(종료:"0")? 0