python - Numpy meshgrid 2d-array
问题描述
我想创建一个网格,我只有一个二维数组。
我想创建这个:
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]])
解决方案
是的,我们可以解包abcd
并传递给messgrid
:
mesh = np.array(np.meshgrid(*abcd))