已知Z = sin(X)+ cos(Y),创建三个相同大小的矩阵(meshgrid),然后将它们绘制为一个曲面,并添加colorbar。
时间: 2024-10-25 17:04:33 浏览: 22
二元函数绘图-rpa之家电商行业解决方案
在Python中,你可以使用`numpy`库的`meshgrid`函数生成二维网格,以及`matplotlib`库来创建图形和颜色条。首先,你需要导入所需的库,然后按照以下步骤操作:
```python
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
# 定义X和Y范围,例如从0到2π
x = y = np.linspace(0, 2 * np.pi, 100) # 创建等差数组,共100个点
# 使用meshgrid函数创建网格
X, Y = np.meshgrid(x, y)
# 计算Z值(这里sin(X) + cos(Y))
Z = np.sin(X) + np.cos(Y)
# 创建3D图形
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
# 绘制曲面
surf = ax.plot_surface(X, Y, Z, cmap='viridis') # 使用viridis色标
# 添加颜色条
cbar = fig.colorbar(surf)
cbar.set_label('Z Value') # 标签颜色条
# 显示图像
plt.title("Surface plot of Z = sin(X) + cos(Y)")
plt.show()
阅读全文