np.meshgrid\
时间: 2023-12-30 13:04:33 浏览: 118
`np.meshgrid()`是一个用于生成网格点坐标矩阵的函数。它接受一系列的一维数组,并返回一个多维数组,其中每个元素都是由输入数组中每个元素的笛卡尔积生成的。下面是一个简单的例子:
```python
import numpy as np
x = np.array([1, 2, 3])
y = np.array([4, 5, 6])
X, Y = np.meshgrid(x, y)
print(X)
print(Y)
```
输出结果为:
```
[[1 2 3]
[1 2 3]
[1 2 3]]
[[4 4 4]
[5 5 5]
[6 6 6]]
```
可以看到,`np.meshgrid()`将输入的一维数组转换成了二维数组,其中第一个数组`X`的每一行都是输入数组`x`的一个副本,而第二个数组`Y`的每一列都是输入数组`y`的一个副本。这样,`X`和`Y`的每个元素就是由`x`和`y`中每个元素的笛卡尔积生成的。
需要注意的是,`np.meshgrid()`的返回值是一个元组,其中每个元素都是一个数组。如果需要将这些数组组合成一个坐标矩阵,可以使用`np.concatenate()`函数。例如:
```python
import numpy as np
x = np.array([1, 2, 3])
y = np.array([4, 5, 6])
X, Y = np.meshgrid(x, y)
coords = np.concatenate([X.reshape(-1, 1), Y.reshape(-1, 1)], axis=1)
print(coords)
```
输出结果为:
```
[[1 4]
[2 4]
[3 4]
[1 5]
[2 5]
[3 5]
[1 6]
[2 6]
[3 6]]
```
这里使用了`np.reshape()`函数将`X`和`Y`转换成了一维数组,并使用`np.concatenate()`函数将它们组合成了一个坐标矩阵。
阅读全文