Adding extra contour lines using Matplotlib 2D contour plotting

Matplotlib Python Data Visualization

Python Data Science basics with Numpy, Pandas and Matplotlib

Data Visualization using MatPlotLib & Seaborn

DATAhill Solutions Srinivas Reddy

More Detail

MatPlotLib with Python

DATAhill Solutions Srinivas Reddy

More Detail

To add extra contour lines using Matplotlib 2D contour plotting, we can take the following steps −

Set the figure size and adjust the padding between and around the subplots.
Create e a function f(x, y) to get the z data points from x and y.
Create x and y data points using numpy.
Make a list of levels using Numpy.
Make a contour plot using contour() method.
Label the contour plot and set the title of the plot.
To display the figure, use show() method.

Example

import matplotlib.pyplot as plt
import numpy as np

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

def f(x, y):
   return np.sin(x) ** 10 + np.cos(10 + y * x) * np.cos(x)

x = np.linspace(0, 5, 50)
y = np.linspace(0, 5, 40)

X, Y = np.meshgrid(x, y)
Z = f(X, Y)
levels = np.arange(-1.0, 1.5, 0.25)
CS = plt.contour(X, Y, Z, levels=levels)

plt.clabel(CS, inline=1, fontsize=10)

plt.title('levels = {}'.format(levels.tolist()))

plt.show()