Plotting bubble plots
Let's see how to plot bubble plots. The size of each circle in a 2D bubble plot represents the amplitude of that particular point.
How to do it…
- Create a new Python file, and import the following packages:
import numpy as np import matplotlib.pyplot as plt
- Define the number of values that we should generate:
# Define the number of values num_vals = 40
- Generate random values for
x
andy
:# Generate random values x = np.random.rand(num_vals) y = np.random.rand(num_vals)
- Define the area value for each point in the bubble plot:
# Define area for each bubble # Max radius is set to a specified value max_radius = 25 area = np.pi * (max_radius * np.random.rand(num_vals)) ** 2
- Define the colors:
# Generate colors colors = np.random.rand(num_vals)
- Plot these values:
# Plot the points plt.scatter(x, y, s=area, c=colors, alpha=1.0) plt.show()
- The full code is in the
bubble_plot.py
file that's already provided to you. If you run this code, you will see the following figure...