How to mark a specific level in a contour map on Matplotlib?


To mark a specific level in a contour map on Matplotlib, we can take the following steps −

  • Set the figure size and adjust the padding between and around the subplots.

  • Create x, y and z data points using Numpy.

  • Use contour() method to make contour plot.

  • Label the contour 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)

contours = plt.contour(X, Y, Z, 3, colors='black')

plt.clabel(contours, inline=True, fontsize=8)

plt.show()

Output

Updated on: 10-Aug-2021

713 Views

Kickstart Your Career

Get certified by completing the course

Get Started
Advertisements