import numpy as np
import scipy.stats as stats
import matplotlib.pyplot as plt

# Rozdělení
n = 20
p = 0.3
b = stats.binom(n, p)

# Jeden obrázek
fig, ax = plt.subplots(1, 1)
x = np.arange(0, n+1)
ax.plot(x, b.pmf(x), 'bo', ms=8, label='binom pmf')
ax.vlines(x, 0, b.pmf(x), colors='b', lw=5, alpha=0.5)
ax.legend()
plt.show()

# Nebo více obrázků pohromadě
fig, ax = plt.subplots(1, 2)
x = np.arange(0, n+1)
ax[0].set_ylim(0, 1)
ax[0].plot(x, b.pmf(x), 'bo', ms=8, label='binom pmf')
ax[0].vlines(x, 0, b.pmf(x), colors='b', lw=5, alpha=0.5)
ax[0].legend()
ax[1].plot(x, b.cdf(x), 'r', label='binom cdf')
ax[1].legend()
plt.show()