Python NumPy dstack方法用法及代碼示例

Numpy 的 dstack(~) 方法用於沿第三軸連接數組。檢查下麵的示例以進行說明。

參數

1. tup | array-like

將沿第三軸連接的數組。

返回值

一個 Numpy 數組，其中提供的數組沿第三軸串聯。

例子

連接一維數組

x = np.array([1,2,3]) 
y = np.array([4,5,6]) 
z = np.dstack([x,y])
z



array([[[1, 4],
        [2, 5],
        [3, 6]]])

請注意，您不限於一次僅連接兩個數組 - 您可以傳入任意數量的數組。另外，這裏返回的 Numpy 數組是三維的。

連接二維數組

x = np.array([[1,2],[3,4]]) 
y = np.array([[5,6],[7,8]]) 
z = np.dstack([x,y])
z



array([[[1, 5],
        [2, 6]],
       [[3, 7],
        [4, 8]]])

為了更容易處理這個問題，這裏是x和y pretty-formatted：

x = [[1, 2],     y = [[5, 6] 
     [3, 4]]          [7, 8]]

相關用法

注：本文由純淨天空篩選整理自Isshin Inada大神的英文原創作品 NumPy | dstack method。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。