finding_max_likelihood.py

import matplotlib.pyplot as plt
import numpy as np

def likelihood(theta):
    return theta**7 * (1 - theta)**3

thetas = np.linspace(0, 1, 100)
L = [likelihood(thetas[i]) for i in range(len(thetas))]

theta_for_maximum_likelihood = thetas[np.argmax(L)]

plt.plot(thetas, L)
plt.xlabel("Values of " + r"$\theta$")
plt.ylabel("Likelihood " + r"$L(y | k=7, n=10; \theta)$")
plt.xticks(list(plt.xticks()[0]) + [theta_for_maximum_likelihood])
plt.axvline(theta_for_maximum_likelihood)
plt.show()