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]]