numpy random choiceの使い方
概要
np.random.choiceは、配列からランダムに要素を選択する関数- seedを固定して再現性を持たせることができる
np.random.seed(42)
- デフォルトでは
replace=Trueで復元抽出
パラメータ
a: 配列size: 選択する要素数replace: 復元抽出 or 非復元抽出p: 選択確率
サンプルコード
復元抽出
import numpy as np
# seedの固定
np.random.seed(42)
data = np.array([1, 2, 3, 4, 5])
sample_with_replacement = np.random.choice(data, size=3, replace=True)
非復元抽出
sample_without_replacement = np.random.choice(data, size=3, replace=False)
選択確率を指定
weights = [0.1, 0.1, 0.2, 0.3, 0.3]
sample_with_weights = np.random.choice(data, size=3, replace=True, p=weights)