random.choicesがうまく動かない

Question

#質問内容
次のプログラムを作成しましたが、関数talkingの第1引数には常に[さようなら]と入力されます。

質問１：正しく3つの文字列を3:3:1の確率で返させるにはどうすればよいでしょうか。
質問２：[]がつかないようにするにはどうすればよいでしょうか。

#プログラムのコード

def index():	
	from robolibs import talking
	import random
	s = str(random.choices(["さようなら", "でわでわ","さよならだけが人生、ですね。"], k=1, cum_weights=[3, 3, 1]))
	talking(s,"normal")

#参考にしたサイト
Pythonでリストからランダムに要素を選択・取得 (note.nkmk.me)

Answer 1

質問２：[]がつかないようにするにはどうすればよいでしょうか。

標準モジュールのrandom.choicesの代わりにnumpy.random.choiceを使うのもありかもしれませんね。

import numpy as np
from numpy import random

p = np.array([3, 3, 1])
p = p / sum(p)  # 確率の総和が1になるよう正規化
random.choice(["さようなら", "でわでわ","さよならだけが人生、ですね。"], p=p)

Comments

[shirasu10fish]

なるほど。こんな方法もあるんですね。試してみます。
ありがとうございました。

Answer 2

キーワード引数cum_weightsは累積的な重みだそうです。
代わりにweightsを指定してはどうでしょうか。

s, *_ = random.choices(
    ["さようなら", "でわでわ","さよならだけが人生、ですね。"], k=1,
    weights=[3, 3, 1]
)

Python 標準ライブラリ » random.choices

---

[]がつかないようにするにはどうすればよいでしょうか。

要素数1のリストが返されるわけですから、先頭要素だけ受け取れば良いのです。

• インデックスを指定する方法

random.choices(略)[0]を受け取れば良いです。

• アンパック記法を用いる方法

上記回答で使っている方法です。慣れていないとわかりづらいかも。

Comments

[shirasu10fish]

無事動きました。ありがとうございました。