【用python画出惊艳的的3d论文图】3D曲面图

在科研绘图中经常需要绘制一些3D曲面图，使用Origin可以达到下面的效果：

但是用Origin自带的功能很难调出令人满意的配色，而其实用python也可以达到与上面类似的效果：

绘制方法如下，首先引入一些包

from matplotlib import cbook
from matplotlib import cm
from matplotlib.colors import LightSource
from mpl_toolkits.axes_grid1 import make_axes_locatable
import matplotlib.pyplot as plt
import numpy as np
import matplotlib as mpl

接着引入数据，这里的数据参考 https://matplotlib.org/stable/gallery/mplot3d/custom_shaded_3d_surface.html#sphx-glr-gallery-mplot3d-custom-shaded-3d-surface-py，实际使用时替换成自己的3维数据就行了。

# Load and format data
dem = cbook.get_sample_data('jacksboro_fault_dem.npz', np_load=True)
z = dem['elevation']
nrows, ncols = z.shape
x = np