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