Theo F - Blatt 2.pdf

Theo F.py

import numpy as np
import matplotlib.pyplot as plt
from scipy.special import binom
from scipy.stats import norm

q = 100
NA = 100
NB = 60

qA = np.arange(0, q + 1, 1)
qB = np.arange(q, -1, -1)

OA = binom(3 * NA - 1 + qA, qA)
OB = binom(3 * NB - 1 + qB, qB)

sm = np.sum(OA * OB)

plt.bar(qA, OA * OB)

mean = np.sum(qA * OA * OB / sm)
variance = np.sum((qA - mean)**2 * OA * OB / sm)

plt.plot(sm * norm.pdf(qA, mean, np.sqrt(variance)), color = 'r')

plt.xlabel("qA")
plt.ylabel("Ω tot")

plt.show()