numpy 库的 meshgrid 函数
时间: 2024-05-15 10:15:54 浏览: 69
numpy 库的 meshgrid 函数用于生成网格点坐标矩阵。其语法如下:
```python
np.meshgrid(*xi, **kwargs)
```
其中,`xi` 表示生成网格点坐标矩阵的坐标轴列表,可以是一个或多个一维数组。`**kwargs` 为可选参数,用于指定网格点坐标矩阵的形状和索引顺序。常用的可选参数有:
- `sparse`:如果为 True,则返回稀疏矩阵格式的网格点坐标矩阵。
- `copy`:如果为 False,则返回的网格点坐标矩阵和输入的坐标轴数组共享内存。
函数返回生成的网格点坐标矩阵,其形状为 `len(xi)` 维,每个维度的大小为 `xi[i].size`。例如,当 `xi=[x,y]` 时,返回的网格点坐标矩阵为 $N_y\times N_x$ 的矩阵,其中 $N_x$ 和 $N_y$ 分别为数组 `x` 和 `y` 的大小。
示例代码:
```python
import numpy as np
x = np.linspace(-1, 1, 3)
y = np.linspace(-1, 1, 3)
X, Y = np.meshgrid(x, y)
print(X)
print(Y)
```
输出结果:
```
[[-1. 0. 1.]
[-1. 0. 1.]
[-1. 0. 1.]]
[[-1. -1. -1.]
[ 0. 0. 0.]
[ 1. 1. 1.]]
```
以上代码生成了一个 $3\times 3$ 的网格点坐标矩阵,其中 $x$ 坐标轴上的点为 $[-1, 0, 1]$,$y$ 坐标轴上的点为 $[-1, 0, 1]$。
阅读全文