matplotlib绘制三维曲面图

import numpy as np
import matplotlib.pyplot as  plt

#绘制一张图
fig=plt.figure()
#绘制3D子图
ax=fig.add_subplot(projection="3d")
x0 = np.arange(-3,3,0.1)
x1 = np.arange(-3,3,0.1)
x0,x1= np.meshgrid(x0, x1)
z = np.array(x0**2 + x1**2)

ax.plot_surface(x0,x1,z,cmap='inferno')
ax.set_xlabel("x0")
ax.set_ylabel("x1")
ax.set_zlabel("f(x)")
ax.set_title("3D")
plt.show()

代码运行结果如下：                

                

 那么我们如何修改colormap(cmap)的值呢？又怎么知道它到底有哪些值呢？

方法一：在python控制台进行以下操作即可得知cmap的参数

方法二：在pycharm中cmap处输入错误参数，错误提示信息中会给出cmap的参数