plt.plot_surface code example

Example 1: 3D plot matplotlib

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

Example 2: Matplotlib surface plotting

from mpl_toolkits import mplot3d