【Python】画3d图(网格优化图)简介

最近找实习有个公司的笔试题,有部分是画3d图,因此学习了一下如何用Python来画3d图。
基本思想:使用matplotlib创建画布,然后使用mpl_toolkits包的Axes3D函数将其转化为3d图画布。

先导入主要用到的函数:

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

然后构造变量三个变量用户作图:

X = np.arange(-2, 2, 0.2)
Y = np.arange(-2, 2, 0.2)
X, Y = np.meshgrid(X, Y)
Z = np.sin(np.sqrt(X**2 + Y**2))

创建画布:

fig = plt.figure()
ax = Axes3D(fig)

画图:

ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=plt.cm.hot)
ax.contourf(X, Y, Z, zdir='z', offset=-2, cmap=plt.cm.hot)
ax.set_zlim(-2,2)
plt.show()

我们下次再见,如果还有下次的话!!!

【新浪微博@516数据工作室】

长按下方二维码关注"516数据工作室"
【Python】画3d图(网格优化图)简介_第1张图片

你可能感兴趣的:(Python)