[파이썬] 여러 카테고리를 일정 확률로 생성

0부터 1 사이의 난수를 발생시키고, 누적확률을 통해 구성한 테이블에 어느 곳에 들어가는가로 확률적 발생 가능

 

 

예시

# 사과, 포도, 복숭아, 귤을 아래와 같은 확률로 100개를 생성
# 사과 20%,  포도 15%,  복숭아 40%,  귤 25%

import random

FRUIT_NUM = 100
# 누적 확률을 과일 순서대로 기록
FRUIT_PROB = [0.20, 0.35, 0.75, 1.00]
FRUIT_DICT = {0: '사과', 1: '포도', 2: '복숭아', 3: '귤'}

fruit_table = [0, 0, 0, 0]
n = random.random()
for i in range(FRUIT_NUM):
	for j in range(4):
		if n < fruit_prob[j]:
    		fruit_table[j] += 1
            break
        
for i in range(4):
	print(f'{FRUIT_DICT[i]}는 {fruit_table[i]}개 있습니다!'}