N = G / G.max()
surf = ax.plot_surface(X, Y, Z, rstride=1, cstride=1, facecolors=cm.jet(N),
    linewidth=0, antialiased=False, shade=False)
plt.show()
Color matplotlib plot_surface command with surface gradient