matplotlib的基本用法(十一)——绘制3D图

文章作者：Tyan
博客：noahsnail.com  |  CSDN  |  简书

本文主要使用matplotlib进行3D图像的绘制。

• Demo

import matplotlib.pyplot as pltimport numpy as npfrom mpl_toolkits.mplot3d import Axes3D# 定义figurefig = plt.figure()# 将figure变为3dax = Axes3D(fig)# 数据数目n = 256# 定义x, yx = np.arange(-4, 4, 0.25)y = np.arange(-4, 4, 0.25)# 生成网格数据X, Y = np.meshgrid(x, y)# 计算每个点对的长度R = np.sqrt(X ** 2 + Y ** 2)# 计算Z轴的高度Z = np.sin(R)# 绘制3D曲面ax.plot_surface(X, Y, Z, rstride = 1, cstride = 1, cmap = plt.get_cmap('rainbow'))# 绘制从3D曲面到底部的投影ax.contour(X, Y, Z, zdim = 'z', offset = -2, cmap = 'rainbow')# 设置z轴的维度ax.set_zlim(-2, 2)plt.show()

• 结果