ax = fig.add_subplot(1, 1, 1, projection='3d')
X, Y = np.meshgrid(np.linspace(-3, 3, 100), np.linspace(-3, 3, 100))
Z = peaks(X, Y)
Z[X + Y > 3] = np.nan
surf = ax.plot_surface(X, Y, Z, alpha=0.3, cmap=plt.cm.jet, vmin=np.nanmin(
    Z), vmax=np.nanmax(Z))
cbar = plt.colorbar(surf)
plt.show()
Как игнорировать NaN в цветной полосе?