How to change the color and add grid lines to a Python Matplotlib surface plot?

To change the color and add grid lines to a Python surface plot, we can take the following steps −

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

  • Create x, y and h data points using numpy.

  • Create a new figure or activate an existing figure.

  • Get 3D axes object, with figure (from Step 3).

  • Create a surface plot, with orange color, edgecolors and linewidth.


import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

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

x = np.arange(-5, 5, 0.25)
y = np.arange(-5, 5, 0.25)

x, y = np.meshgrid(x, y)
h = np.sin(x)*np.cos(y)

fig = plt.figure()

ax = Axes3D(fig)
ax.plot_surface(x, y, h, rstride=10, cstride=10, color='orangered', edgecolors='yellow', lw=0.6)


Updated on: 23-Sep-2021

1K+ Views

Kickstart Your Career

Get certified by completing the course

Get Started