Check N(mu, sigma) distribution visually.

gauss_dist_checker.py

#!/usr/bin/env python3

import matplotlib.pyplot as plt
from random import gauss

NUM_SAMPLES = 128


def plot_gaussian_distribution(mu: float, sigma: float) -> None:
    """Check N(mu, sigma) distribution visually."""
    nums = [gauss(mu, sigma) for _ in range(NUM_SAMPLES)]
    # Plot histogram
    plt.hist(nums)
    plt.show()


def get_user_input(prompt: str) -> float:
    return float(input(f'\033[93m{prompt} >> \033[0m'))


if __name__ == '__main__':
    m = get_user_input("mu?")
    s = get_user_input("sigma?")
    plot_gaussian_distribution(m, s)