matplotlib_object_oriented_line_plot_with_legend.py

1import matplotlib.pyplot as plt
2import numpy as np
3
4# Data for plotting
5x = np.linspace(0, 2, 100)
6
7# Note that even in the OO-style, we use `.pyplot.figure` to create the Figure.
8fig, ax = plt.subplots(figsize=(5, 2.7), layout='constrained')
9ax.plot(x, x, label='linear')  # Plot some data on the axes.
10ax.plot(x, x**2, label='quadratic')  # Plot more data on the axes...
11ax.plot(x, x**3, label='cubic')  # ... and some more.
12ax.set_xlabel('x label')  # Add an x-label to the axes.
13ax.set_ylabel('y label')  # Add a y-label to the axes.
14ax.set_title("Simple Plot")  # Add a title to the axes.
15ax.legend()  # Add a legend.
16
17plt.show()