How to draw axis in the middle of a figure in Matplotlib?

To draw axis in the middle of a figure, we can take the following steps −

• Create x and sqr data points using numpy.

• Create a new figure, or activate an existing figure, using figure() method.

• Add an axis to the figure as a part of a subplot arrangement.

• Set the postion of left and bottom spines.

• Set the color of the right and top spines.

• Plot x and sqr, using plot() method, with label y=x2 and color=red.

• Place the legend using legend() method. Set the location at upper right corner.

• To display the figure, use show() method.

Example

import numpy as np import matplotlib.pyplot as plt plt.rcParams["figure.figsize"] = [7.00, 3.50] plt.rcParams["figure.autolayout"] = True x = np.arange(-10., 10., 0.2) sqr = np.square(x) fig = plt.figure() ax = fig.add_subplot(1, 1, 1) ax.spines['left'].set_position('center') ax.spines['bottom'].set_position('center') ax.spines['right'].set_color('none') ax.spines['top'].set_color('none') plt.plot(x, sqr, label="y=x^2", c='red') plt.legend(loc=1) plt.show()

Output