plot_surface参数
时间: 2023-10-18 21:04:46 浏览: 162
`plot_surface()` 是 Matplotlib 中用于绘制三维表面图的函数。它的常用参数如下:
- `X`, `Y`, `Z`: 数组,表示三维数据点的坐标。
- `rstride`, `cstride`: 可选参数,表示行和列的跨度(即数据点之间的间隔)。默认值为 1。
- `cmap`: 可选参数,表示颜色图谱,用于给表面图着色。默认值为 `'viridis'`。
- `edgecolor`: 可选参数,表示网格线的颜色。默认值为 `'none'`。
- `linewidth`: 可选参数,表示网格线的宽度。默认值为 0.5。
- `antialiased`: 可选参数,表示是否使用抗锯齿。默认值为 `True`。
示例代码:
```python
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
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)
surf = ax.plot_surface(X, Y, Z, cmap='coolwarm', edgecolor='none')
ax.set_title('Surface plot')
fig.colorbar(surf, shrink=0.5, aspect=5)
plt.show()
```
这段代码使用了 `plot_surface()` 函数绘制了一个三维表面图。首先定义了 X、Y、Z 三个数组,通过 `np.meshgrid()` 函数生成了网格点坐标。然后用 `np.sqrt()` 和 `np.sin()` 函数计算了每个点的高度值 Z。最后调用 `plot_surface()` 函数绘制了表面图,并通过 `set_title()` 函数设置了标题,通过 `colorbar()` 函数添加了颜色条。
阅读全文