玉の出現確率を調べたい

Question

前提・実現したいこと

ある袋の中に入っている５つの玉「赤玉」「青玉」「黒玉」「緑玉」「黄玉」をランダムに１つ取り出して玉の色を確認し、
取り出した玉は確認後にもとに戻してもう一度、ランダムに１つ取り出す作業を1000回繰り返したときに
それぞれの玉の出現する確率をRubyにて調べたい。

該当のソースコード

balls = ["赤玉", "青玉", "黒玉", "緑玉", "黄玉"]
count = [0, 0, 0, 0, 0]

1000.times do |i|
  if balls.sample == "赤玉"
    count[0] += 1
  elsif balls.sample == "青玉"
    count[1] += 1
  elsif balls.sample == "黒玉"
    count[2] += 1
  elsif balls.sample == "緑玉"
    count[3] += 1
  else
    count[4] += 1
  end
end

5.times do |i|
   printf("%s: %.1f%\n", balls[i], count[i] / 1000.0 * 100)
end

どこがおかしいか教えていただきたいです。

Answer 1

ifやelsifの中で毎回balls.sampleを実行しているため、判定条件が場所ごとに違うことになってしまいます。最初にballs.sampleの結果を変数に入れて、それと比較を行いましょう。

Comments

[xyjs]

詳細に説明いただき、ありがとうございます。

Answer 2

balls.sample の結果を ball という変数に格納させたら解決しました。

balls = ["赤玉", "青玉", "黒玉", "緑玉", "黄玉"]
count = [0, 0, 0, 0, 0]

1000.times do |i|
  ball = balls.sample
  if ball == "赤玉"
    count[0] += 1
  elsif ball == "青玉"
    count[1] += 1
  elsif ball == "黒玉"
    count[2] += 1
  elsif ball == "緑玉"
    count[3] += 1
  else
    count[4] += 1
  end
end

5.times do |i|
   printf("%s: %.1f%\n", balls[i], count[i] / 1000.0 * 100)
end

Answer 3

balls.sample　の code がないので推測ですが、
一回ごとに乱数で色を決めていますか？

だとすると今のプログラmは、合計回数が1000にならず、360前後になってしまうでしょう。

5n回めが赤、5n+1回目が青、5n+2回目が黒,5n+3回目が緑であった時にカウントアップし、5n+4回目は何が来ても黄をカウントアップする
ということになってしまってます。
if の前に color = balls.sample して、colorを確認するようにしましょう

追記
私はifが嫌いなので、こんな風にします

count = Hash.new{|h,k| h[k] = 0}
1000.times { count[balls.sample] += 1 }

Comments

[maisumakun]

> balls.sample　の code がないので

Array#sampleはRuby標準であるメソッドです。

https://docs.ruby-lang.org/ja/latest/method/Array/i/sample.html

[winterboum]

お恥ずかしい

[asm]

tally便利ですよ
counts = Array.new(1000){balls.sample}.tally
# ↓配列を作りたくない場合
counts = Enumerator.produce{balls.sample}.lazy.take(1000).tally

[xyjs]

様々な書き方があるのですね。
説明いただき、ありがとうございます。