Numpy meshgrid 二维数组

Question

我想创建一个网格，但我只有一个二维数组。

我想创建这个：

a = np.array([1,2])
b = np.array([3,4])
mesh = np.array(np.meshgrid(a, b))
## mesh = 
## array([[[1, 2],
##    [1, 2]],
##
##   [[3, 3],
##    [4, 4]]])

但是输入是一个包含a和b的数组，像这样：

ab = np.array([[1,2],[3,4]])
mesh = np.array(np.meshgrid(ab))
## mesh = 
## array([[[1, 2],
##    [1, 2]],
##
##   [[3, 3],
##    [4, 4]]])

即使将 3 或 4 个数组打包成一个像这样的大数组，他们是否也是一种存档方式：

abcd = np.array([[1,2],[3,4],[5,6],[7,8]])

Answer 1

是的，我们可以解压 abcd 并传递给 messgrid:

mesh = np.array(np.meshgrid(*abcd))

Numpy meshgrid 二维数组

Numpy meshgrid 2d-array

python

arrays

numpy

mesh