What is the equivalent of Matlab's surf(x,y,z,c) in Matplotlib?

Let's take an example to see how to get the same effect as MatLab's surf(x,y,z,c) in Matplotlib. steps −

Set the figure size and adjust the padding between and around the subplots.
Create a new figure or activate an existing figure.
Add an 'ax' to the figure as part of a subplot arrangement.
Create r, u, v, x, y and z data points using Numpy.
Create a surface 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

fig = plt.figure()
ax = fig.add_subplot(projection='3d')
r = 0.05
u, v = np.mgrid[0:2 * np.pi:30j, 0:np.pi:20j]

x = np.cos(u) * np.sin(v)
y = np.sin(u) * np.sin(v)
z = np.cos(v)

ax.plot_surface(x, y, z, cmap=plt.cm.YlGnBu_r)

plt.show()

Output

Rishikesh Kumar Rishi

Updated on: 2021-10-21T08:48:15+05:30

876 Views

Kickstart Your Career

Get certified by completing the course

Get Started