PythonのNumpyを用いてガチャシュミレータを作成したい

Question

PythonのNumpyライブラリにある乱数(random)を使って次のことをしたいです。

Aが1%、Bが5%、Cが8%、Dが86%の確率で発生する乱数を作成し、100回抽選を行ったときに、それぞれ何回発生したかを表示するプログラムを作成したいのですが、どのように作成すればよいか分からず、困っています。

よろしければ教えてください。

Answer 1

以下のようにして可能です。

1. numpy.random.choice(アイテム一覧, 抽選回数, 確率一覧) で指定した確率に従い、100回抽選を行う
2. numpy.unique(抽選結果, return_counts=True) で各アイテムの出現数を数える

numpy.random.choice — NumPy v1.15 Manual
numpy.unique — NumPy v1.19 Manual

import numpy as np

items = ["A", "B", "C", "D"]  # アイテム一覧
prob = [0.01, 0.05, 0.08, 0.86]  # 確率一覧 (総和が1になっていること)
N = 100  # 抽選回数

# 抽選を行う
result = np.random.choice(items, N, p=prob)
print(result)
# ['D' 'B' 'D' 'D' 'D' 'C' 'D' 'D' 'D' 'C' 'D' 'C' 'D' 'D' 'D' 'D' 'D' 'D'
#  'D' 'D' 'D' 'B' 'D' 'D' 'C' 'D' 'C' 'D' 'D' 'D' 'C' 'B' 'D' 'D' 'D' 'D'
#  'D' 'C' 'D' 'D' 'D' 'D' 'D' 'D' 'B' 'D' 'D' 'D' 'D' 'D' 'D' 'D' 'D' 'D'
#  'D' 'D' 'D' 'B' 'D' 'D' 'C' 'D' 'D' 'D' 'D' 'D' 'D' 'D' 'D' 'D' 'D' 'D'
#  'D' 'D' 'D' 'D' 'C' 'D' 'D' 'D' 'D' 'B' 'D' 'D' 'D' 'D' 'D' 'D' 'D' 'D'
#  'D' 'D' 'D' 'D' 'D' 'D' 'D' 'C' 'D' 'D']

# 集計する
items, counts = np.unique(result, return_counts=True)

for item, count in zip(items, counts):
    print(item, count)
# A 2
# B 5
# C 8
# D 85

Comments

[fafw]

わかりやすく回答して頂き有難うございます。
ベストアンサーにさせて頂きました。