Numpy 網格二維陣列

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 個 arrays 打包成這樣一個大文件：

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

Answer 1

是的，我們可以解壓abcd並傳遞給messgrid ：

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