📂프로그래밍

파이썬에서 복원추출, 비복원추출하는 방법

코드

파이썬에서 복원/비복원추출을 하는 방법은 numpy의 함수 np.random.choice()를 사용하면 된다.

random.choice(a, size=None, replace=True, p=None)

• a: 1차월 배열이나 정수
  • 추출을 진행할 집합을 의미한. a가 정수일 경우 np.arange(a)에서 추출된다.
• size: 정수 혹은 정수들의 튜플
  • 추출한 출력의 사이즈를 의미한다.
• replace: 불리언
  • T는 복원추출, F는 비복원추출을 의미한다.
• p: 1차월 배열
  • 각 원소가 뽑힐 확률을 의미한다. 기본값은 None이다.

(23.10.04)파이토치로는 간단하게 할 수 있는 방법이 없다.

>>> np.random.choice(5, 5, replace=False)
array([3, 1, 0, 2, 4])

>>> np.random.choice(5, 5, replace=True)
array([0, 2, 0, 2, 2])

환경

• OS: Windows11
• Version: Python 3.11.5, numpy==1.26.0