SciPy Recipes

Visualizing the probability mass function

pmf gives us an output of the mass (probability) associated with each likely output. In the example that we looked at earlier, we have the probability associated with each value of x (although the probabilities vary depending on the value of x).

In this section, we will look at plotting the probability mass function for each possible value of x:

1. Import the relevant packages for plotting:

import matplotlib.pyplot as plt

2. Initialize the plots:

fig, ax = plt.subplots(1, 1)

3. Plot the values of x along with the probability mass function associated with each value of x:

ax.plot(x, binom.pmf(x, n, p), 'bo', ms=8, label='binom pmf')[<matplotlib.lines.Line2D at 0x681f474518>]

4. Show the plot:

plt.show() ...