Python list에서 랜덤하게 아이템 꺼내기

테스트용 데이터 넣다가 유용하게 사용할 듯 해서 정리함

100개의 사이트가 있다고 가정하고 100만개의 데이터가 100개의 사이트를 랜덤하게 갖는 테스트 데이터를 생성하려고 한다.

100만번의 루프를 돌 때 각 루프에서 테스트용 100개의 사이트 중 랜덤하게 하나의 사이트를 골라서 DB에 넣고 싶었다

import random
sites = ['site-{0}'.format(i) for i in range(0, 100)]
for i in range(0, 1000000):
    print (random.choice(sites))

결과는

/Library/Frameworks/Python.framework/Versions/3.5/bin/python3.5 /Users/airkjh/PycharmProjects/study/etc/random_choice.py
site-75
site-94
site-8
site-85
site-80
site-64
site-61
site-65
site-40
site-66
site-85
site-29
site-24
site-65
site-19
site-7
...

DB에서 쿼리해봤더니 의도한대로 입력이 되었다

select siteid, count(*) from nodes group by siteid

답글 남기기

아래 항목을 채우거나 오른쪽 아이콘 중 하나를 클릭하여 로그 인 하세요:

WordPress.com 로고

WordPress.com의 계정을 사용하여 댓글을 남깁니다. 로그아웃 /  변경 )

Google+ photo

Google+의 계정을 사용하여 댓글을 남깁니다. 로그아웃 /  변경 )

Twitter 사진

Twitter의 계정을 사용하여 댓글을 남깁니다. 로그아웃 /  변경 )

Facebook 사진

Facebook의 계정을 사용하여 댓글을 남깁니다. 로그아웃 /  변경 )

%s에 연결하는 중