Pythonによるバンディット問題のプログラム

Question

Pythonを使用してバンディット問題のシュミレータプログラムを書いています。
http://qiita.com/mzmttks/items/f9a38274518bf576c19c
上記のURLを参考にしています。

冒頭部分の
def getResult(prob):
return scipy.stats.bernoulli.rvs(prob, size=1)[0]
この部分が何を行っているコードなのかわかりません。

関数getResultを作り、引数をprobとし、戻り値として scipy.stats.bernoulli.rvs(prob, size=1)[0]としているのはわかるのですが、rvs(prob, size=1)[0]がわかりません。
戻り値で何が帰ってくるのかが特にわからないです。

解説お願いいたします。

Answer 1

rvsでは、ある確率(prob)で1を出力するものです。
size=1で指定しているのは、その試行を何回行うかです。

例えば、prob=0.1, size=10にすると、サイズ10の配列の各要素に、10%の確率で1が入ります。
簡単に言うと、10個中1個があたりのくじ(= prob)を10回(= size)引いたとき、当たったらそこに1を入れる感じです。
出力が配列のため、[0]で一つ目の要素をとってきています。
（rvsはsize指定しなかった場合、intで帰ってきた気もしますが、versionが違うのかもしれません）

つまり、getResult関数では、引数のprobの確率で当たるくじを一回試行して、当たったか外れたかを1or0で返します。