pythonでガチャを作成したい

Question

pythonでガチャを実装

pythonでガチャを次のように実装してみました。

import random

for i in range(10):
    n=random.randint(0,99)
    if n < 10:
        print("１等")
    elif n < 50:
        print("２等")
    else:
        print("３等")

##実装したいこと
このガチャはいちいち全ての場合を書き出さないといけません。なので、確率と排出物を決めるとガチャにセットできるような仕組みがあれば教えて欲しいです。

Answer 1

numpy を使うやり方

numpy.random.choice() に確率、排出物、回数を指定すると、結果が得られます。

numpy.random.choice — NumPy v1.15 Manual

import numpy as np

items = ["1等", "2等", "3等"]  # 内容
prob = [0.1, 0.3, 0.6]  # 確率
N = 10  # 回数

result = np.random.choice(items, N, p=prob)
print(result)
# ['2等' '3等' '3等' '2等' '3等' '2等' '2等' '3等' '3等' '2等']

random モジュールを使うやり方

random.choices() でもできます。

random.choices(population, weights=None, *, cum_weights=None, k=1)

• population: 選択するアイテム一覧
• weights: 各アイテムを選択する確率
• k: 選択する数

random --- 擬似乱数を生成する — Python 3.8.4rc1 ドキュメント

import random

items = ["1等", "2等", "3等"]  # 内容
prob = [0.1, 0.3, 0.6]  # 確率
N = 10  # 回数

n = random.choices(items, weights=prob, k=N)
print(n)
# ['3等', '1等', '2等', '3等', '1等', '2等', '2等', '3等', '3等', '2等']

Comments

[padms0206]

ありがとうございます！
参考にしてみます！

[tiitoi]

もし numpy を使わないで random モジュールのみを使った方法を質問したのであれば、追記しますので、教えてください

[padms0206]

できれば、その方法もあれば教えて欲しいです。

[tiitoi]

追記しました。random.choices() を使えばできます。

[padms0206]

ありがとうございます！