同じコインを複数回投げた結果から、どのコインだったかの確率

# coin_probability2.rb
def main
  # πi: それぞれのコインが箱の中にどのような割合で混ざっているか
  pi = [0.1, 0.4, 0.5]

  # θi: それぞれのコインを投げた時に表が出る確率
  theta = [0.8, 0.6, 0.3]

  n = 10
  (0..n).each do |r|
    puts "*** #{r}"
    calc_posterior_probability_n(pi, theta, n, r).each_with_index do |post_prob, i|
      puts "P(omega_#{i+1} | H) = #{post_prob}"
    end
  end
end

# 箱の中から取り出したコインをn回投げてr回表が出た時に、
# 取り出したコインがどれだったか、それぞれのコインの確率を求める
# _pi_ :: それぞれのコインが箱の中にどのような割合で混ざっているか、の配列
# _theta_ :: それぞれのコインを投げた時に表が出る確率、の配列
# _n_ :: 試行回数
# _r_ :: 表が出た回数
def calc_posterior_probability_n(pi, theta, n, r)
  c = pi.size
  # P(x^(n)): コインをn回投げてr回表が出たという、順序ありの観測結果x^(n)が得られる確率
  head_prob = sigma(0...c) do |i|
    pi[i] * (theta[i] ** r) * ((1 - theta[i]) ** (n - r))
  end

  # それぞれのコインが取り出された確率を求める
  (0...c).map do |i|
    (pi[i] * (theta[i] ** r) * ((1 - theta[i]) ** (n - r))) / head_prob
  end
end

# Rangeに対してブロックを呼び出し、その結果の和を返す：∑
def sigma(range, &block)
  range.map do |i|
    block.call(i)
  end.inject(:+)
end

main