matplotlib绘制三维曲面图

2022-12-11 02:59| 来源: 网络整理| 查看: 265

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的参数

【本文地址】

公司简介

联系我们