使用matplotlib绘制三维图形
程序员文章站
2022-03-08 14:38:57
...
一 代码
import numpy as np
import matplotlib.pyplot as plt
import mpl_toolkits.mplot3d
x,y = np.mgrid[-2:2:20j,-2:2:20j]
#测试数据
z=x*np.exp(-x**2-y**2)
#三维图形
ax = plt.subplot(111, projection='3d')
ax.plot_surface(x,y,z,rstride=2, cstride=1, cmap=plt.cm.Blues_r)
#设置坐标轴标签
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')
plt.show()
二 运行结果
三 代码
import pylab as pl
import numpy as np
import mpl_toolkits.mplot3d
rho, theta = np.mgrid[0:1:40j,0:2*np.pi:40j]
z = rho**2
x = rho*np.cos(theta)
y = rho*np.sin(theta)
ax = pl.subplot(111, projection='3d')
#ax.plot_surface(x,y,z)
ax.plot_surface(x,y,z,rstride=2, cstride=1)
#设置坐标轴标签
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')
pl.show()
四 运行结果
上一篇: python 登录装饰器