import scipy.stats as stats

# Pořídíme si binomiální rozdělení s danými parametry
n = 20
p = 0.3
b = stats.binom(n, p)

# Necháme si vygenerovat náhodný vzorek
print('RVS:', b.rvs(size=20))

# Pravděpodobnostní funkce (Probability Mass Function)
print('PMF:', [b.pmf(x) for x in range(0, n+1)])

# Distribuční funkce (Cummulative Distribution Function)
print('CDF:', [b.cdf(x) for x in range(0, n+1)])

# Kvantily ("inverze" k CDF, Percent Point Function)
print('Quantile 0.1:', b.ppf(0.1))
print('Quantile 0.9:', b.ppf(0.9))

# Střední hodnota, rozptyl a směrodatná odchylka
print('Mean:', b.mean())
print('Variance:', b.var())
print('Std deviation:', b.std())

# Také se jde přímo zeptat na vlastnosti rozdělení s konrétními
# parametry, aniž bychom ho explicitně konstruovali.
print(stats.binom.rvs(n, p, size=20))