meshgrid 使用
时间: 2023-07-11 08:14:39 浏览: 72
`meshgrid` 是一个 numpy 库中的函数,用于生成一个二维的坐标矩阵。假设我们有两个一维数组 x 和 y,其中 x 包含 m 个元素,y 包含 n 个元素,那么 `meshgrid` 函数可以生成两个矩阵 X 和 Y,它们的形状都是 (n, m),其中 X 的每一行都是 x 数组的复制,Y 的每一列都是 y 数组的复制,例如:
```python
import numpy as np
# 生成一个一维数组 x
x = np.linspace(-5, 5, 11)
# 生成一个一维数组 y
y = np.linspace(-3, 3, 7)
# 使用 meshgrid 生成二维矩阵 X 和 Y
X, Y = np.meshgrid(x, y)
# 打印 X 和 Y
print(X)
print(Y)
```
运行上面的代码,可以得到如下的输出:
```
[[-5. -4. -3. -2. -1. 0. 1. 2. 3. 4. 5.]
[-5. -4. -3. -2. -1. 0. 1. 2. 3. 4. 5.]
[-5. -4. -3. -2. -1. 0. 1. 2. 3. 4. 5.]
[-5. -4. -3. -2. -1. 0. 1. 2. 3. 4. 5.]
[-5. -4. -3. -2. -1. 0. 1. 2. 3. 4. 5.]
[-5. -4. -3. -2. -1. 0. 1. 2. 3. 4. 5.]
[-5. -4. -3. -2. -1. 0. 1. 2. 3. 4. 5.]]
[[-3. -3. -3. -3. -3. -3. -3. -3. -3. -3. -3.]
[-1. -1. -1. -1. -1. -1. -1. -1. -1. -1. -1.]
[ 1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
[ 3. 3. 3. 3. 3. 3. 3. 3. 3. 3. 3.]
[ 5. 5. 5. 5. 5. 5. 5. 5. 5. 5. 5.]
[ 7. 7. 7. 7. 7. 7. 7. 7. 7. 7. 7.]]
```
可以看出,X 矩阵的每一行都是 x 数组的复制,而 Y 矩阵的每一列都是 y 数组的复制。这个二维坐标矩阵可以用于绘制二维图形,如曲面图等。
阅读全文