[Python] 로또 번호 생성

Python3 기준으로 로또 번호 생성 함수를 만들어 보았다.
굳이 shuffle을 하지 않아도 random으로 뽑히기는 하지만
실제 공을 추첨할 때 섞으면서 뽑으니까... 비슷하게 구현해 보았다.

 

#!/usr/bin/env python3

from random import random, shuffle

def lotto(total, pick):
    nums = list(range(1, total+1))
    picked_nums = []
    for i in range(pick):
        shuffle(nums)
        picked_index = int(random() * len(nums))
        picked_num = nums.pop(picked_index)
        picked_nums.append(picked_num)
    return picked_nums

# 로또 공 45개 중 6개를 뽑아보자
if __name__ == '__main__':
    print(lotto(45, 6))