Python使用matplotlib绘制三维曲面图的两种方法

使用matplotlib绘制三维图形

在Python中，利用 matplotlib 库可以方便地实现三维图形的绘制。以下通过两个示例代码及其运行结果，展示具体的实现过程。

示例一：基于笛卡尔坐标的三维曲面绘制

该示例通过生成测试数据，绘制了一个三维曲面图。代码中使用了 np.mgrid 生成网格数据，并通过 plot_surface 函数绘制曲面。关键参数包括 rstride 和 cstride，用于控制行和列的步长，以及 cmap 用于设置颜色映射。

# coding=utf-8
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.set_title('www.jb51.net - matplotlib Demo');
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()

示例二：基于极坐标的三维曲面绘制

第二个示例展示了如何基于极坐标绘制三维图形。通过生成极坐标数据，转换为笛卡尔坐标后，使用 plot_surface 函数绘制曲面。代码中同样设置了坐标轴标签，并展示了运行结果。

# coding=utf-8
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.set_title('www.jb51.net - matplotlib Demo');
#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基于matplotlib实现绘制三维图形功能示例的详细内容，更多关于Python matplotlib三维图形的资料请关注本站其它相关文章！