# パッケージがインストール済なら次の行は不要
# import Pkg; Pkg.add("Plots")
using Plots  # グラフ描画のためのモジュール
pyplot()
# 2人の子供の性別がどちらも男である確率(理論値)を求める関数
function prob_twoboys(pb, pg)
    cond_prob = 0.0
    prob_num = pb .* (2.0 .- pb)
    prob_denom = (pb .+ pg) .* (4.0 .- pb .- pg)
    cond_prob = prob_num ./ prob_denom
    return cond_prob
end
# 2人の子供の性別がどちらも男である確率(理論値)
println(prob_twoboys(0.05, 0))
println(prob_twoboys(0.05, 0.001))
println(prob_twoboys(0.3, 0))
println(prob_twoboys(0.3, 0.01))
# 3次元グラフを描画
pb = LinRange(0, 1, 51)
pg = LinRange(0, 1, 51)
PB = [x for y in pg, x in pb]
PG = [y for y in pg, x in pb]
prob_bb = prob_twoboys.(PB, PG)
fig = surface(pb, pg, prob_bb,
              color=:Blues, alpha=0.7, linecolor=:black, 
              xlabel="pb", ylabel="pg", zlabel="prob_bb",
              camera=(30, 40))
display(fig)