# plotBinomial.py
# inputs: 50 0.5

import matplotlib.pyplot as plt 
from binomial import *

n, p = map(float, input("n, p=? ").split())
n = int(n)
xs = list(range(n+1)) 
dist = [binomial(n, p, x) for x in xs ] 
plt.bar(xs, dist) 
plt.xlabel("n")
plt.ylabel("prob")
plt.title("Binomial Distribution; n="+str(n)+", p="+str(p))
plt.show()