【NumPy】配列から要素のランダム抽出

Python

2017.09.16

この記事では、Python言語とNumPyを用いて配列から要素をランダム抽出する方法をソースコード付きで解説します。

## 配列から要素のランダム抽出

NumPy配列では、numpy.random.choiceで配列から要素をランダム抽出できます。

書式

b = numpy.random.choice(a, n, replace=True)

配列aから要素n個をランダムに抽出します。
（replaceがTrueなら重複あり)

b = numpy.random.choice(a, n, p= [0.1, 0.9])

各要素の出現確率pを指定して配列aから要素n個をランダムに抽出します。

## ソースコード

サンプルプログラムのソースコードです。

# -*- coding: utf-8
import numpy as np


# 配列生成
a = np.array([1, 2, 3])

# 配列のランダム抽出
c1 = np.random.choice(a, 3, replace=True) # 重複あり
c2 = np.random.choice(a, 3, replace=False) # 重複なし
c3 = np.random.choice(a, 3,  p=[0.1, 0.8, 0.1]) # 出現率pを指定

# 結果表示
print(c1) # [2 3 2]
print(c2) # [1 2 3]
print(c3) # [2 2 2]