python画三维曲面图
时间: 2023-08-30 12:05:26 浏览: 103
基于Python PyQt5实现绘制三维曲面图项目源码
5星 · 资源好评率100%
要画三维曲面图,可以使用Python的Matplotlib库中的mplot3d模块。以下是一个简单的例子:
```python
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
# 创建数据
x = np.arange(-5, 5, 0.25)
y = np.arange(-5, 5, 0.25)
x, y = np.meshgrid(x, y)
r = np.sqrt(x ** 2 + y ** 2)
z = np.sin(r)
# 创建图像和坐标轴
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
# 绘制曲面图
surf = ax.plot_surface(x, y, z, cmap='coolwarm')
# 添加标签和标题
ax.set_xlabel('X Label')
ax.set_ylabel('Y Label')
ax.set_zlabel('Z Label')
ax.set_title('3D Surface Plot')
# 显示图像
plt.show()
```
这个例子中,我们创建了一个半径为r的圆柱体,然后把它的高度设为sin(r)。然后我们使用meshgrid创建了网格,然后使用plot_surface方法绘制了曲面。最后,我们添加了标签和标题,并使用show()方法显示图像。
阅读全文