Open in App
Not now

# 3D Contour Plotting in Python using Matplotlib

• Last Updated : 21 Apr, 2020

Matplotlib was introduced keeping in mind, only two-dimensional plotting. But at the time when the release of 1.0 occurred, the 3d utilities were developed upon the 2d and thus, we have 3d implementation of data available today! The 3d plots are enabled by importing the mplot3d toolkit. Let’s look at a 3d contour diagram of a 3d cosine function. The code is attached for reference.

Function Arguments Description
meshgrid Function of numpy used to create a rectangular grid out of two given one-dimensional arrays representing the Cartesian indexing or Matrix indexing
plt.axes() To create object of the axis
ax.contour3D To form the contour
ax.set_xlabel To label the X-axis
ax.set_title To give a title to the plot

Example 1:

 from mpl_toolkits import mplot3d import numpy as np import matplotlib.pyplot as plt from matplotlib import cm import math    x = [i for i in range(0, 200, 100)] y = [i for i in range(0, 200, 100)]    X, Y = np.meshgrid(x, y) Z = [] for i in x:     t = []     for j in y:         t.append(math.cos(math.sqrt(i*2+j*2)))     Z.append(t)    fig = plt.figure() ax = plt.axes(projection='3d') ax.contour3D(X, Y, Z, 50, cmap=cm.cool) ax.set_xlabel('a') ax.set_ylabel('b') ax.set_zlabel('c') ax.set_title('3D contour for cosine') plt.show()

Output:

Example 2: Let’s look at another 3d diagram for better understanding of the concept. This time, of a 3d tan function.

 from mpl_toolkits import mplot3d import numpy as np import matplotlib.pyplot as plt from matplotlib import cm import math    x = [i for i in range(0, 200, 100)] y = [i for i in range(0, 200, 100)]    X, Y = np.meshgrid(x, y) Z = [] for i in x:     t = []     for j in y:         t.append(math.tan(math.sqrt(i*2+j*2)))     Z.append(t)    fig = plt.figure() ax = plt.axes(projection='3d') ax.contour3D(X, Y, Z, 50, cmap=cm.cool) ax.set_xlabel('a') ax.set_ylabel('b') ax.set_zlabel('c') ax.set_title('3D contour for tan') plt.show()

Output:

My Personal Notes arrow_drop_up
Related Articles