NumPy vstack() 函式

NumPy 的 vstack() 函式用於垂直（按行）堆疊陣列。它接受一系列陣列，並沿垂直軸（即軸 0）連線它們。

此函式特別適用於組合列數相同但行數不同的陣列。要堆疊的陣列應具有相同的列數。

例如，如果我們有兩個形狀為 (2, 3) 和 (3, 3) 的二維陣列，vstack() 將將其組合成一個形狀為 (5, 3) 的陣列。該函式返回一個新陣列，並且不修改原始陣列。

語法

NumPy vstack() 函式的語法如下：

numpy.vstack(tup)

引數

NumPy vstack() 函式接受一個引數，即 tup，它是一系列一維或二維陣列。所有陣列都必須沿除第一個軸之外的所有軸具有相同的形狀。一維陣列必須具有相同的長度。

返回值

此函式返回一個透過垂直堆疊給定陣列形成的二維陣列。

示例 1

以下是 NumPy vstack() 函式的示例，它演示瞭如何垂直堆疊兩個一維陣列以形成一個二維陣列：

import numpy as np

array1 = np.array([1, 2, 3])
array2 = np.array([4, 5, 6])

result = np.vstack((array1, array2))
print(result)

輸出

[[1 2 3]
 [4 5 6]]

示例 2

以下示例演示瞭如何在形狀與被堆疊軸相容的情況下，將一維陣列與二維陣列堆疊：

import numpy as np

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

result = np.vstack((array1, array2))
print(result)

輸出

[[1 2 3]
 [4 5 6]
 [7 8 9]]