Function expectation

maths/joint_probability_distribution.py:31–37

>>> from math import isclose >>> isclose(expectation([1, 2], [0.7, 0.3]), 1.3) True

(values: list, probabilities: list)

# Function to calculate the expectation (mean)
def expectation(values: list, probabilities: list) -> float:
    """
    >>> from math import isclose
    >>> isclose(expectation([1, 2], [0.7, 0.3]), 1.3)
    True
    """
    return sum(x * p for x, p in zip(values, probabilities))


# Function to calculate the variance