# -*- coding: utf-8 -*-
"""
Created on Wed Jun 23 15:22:31 2021

@author: Sim
"""
from scipy.stats import binom
from scipy.stats import norm
import numpy as np
import matplotlib.pyplot as plt

nn = 15
p1 = 0.7; p2 = 0.5
mu1 = nn*p1; mu2 = nn*p2
sigma1 = (nn*p1*(1-p1)) ** .5;
sigma2 = (nn*p2*(1-p2)) ** .5;

x = list(range(nn))
x2 = np.linspace(0,nn, 100)
pmf1 = binom.pmf(x, nn, p1)
pdf1 = norm.pdf(x2, mu1, sigma1)
pmf2 = binom.pmf(x, nn, p2)
pdf2 = norm.pdf(x2, mu2, sigma2)

plt.bar(x, pmf1, label='B(15,0.7)')
plt.plot(x2, pdf1, 'r-', label='N(10.5,3.15)')
plt.legend()
plt.show()

plt.bar(x, pmf2, label='B(15,0.5)')
plt.plot(x2, pdf2, 'r-', label='N(7.5,3.75)')
plt.legend()
plt.show()