np.vstack() 函数是 NumPy 中用来按垂直方向(沿着行方向)堆叠数组的函数。它将两个或多个数组沿着垂直方向堆叠在一起,要求这些数组在水平方向上的维度相匹配。
下面是一个简单的示例,说明了 np.vstack() 函数的用法:
示例1:
import numpy as np
# 创建两个示例数组
array1 = np.array([[1, 2, 3],
[4, 5, 6]])
array2 = np.array([[7, 8, 9],
[10, 11, 12]])
# 使用 np.vstack() 函数沿垂直方向堆叠这两个数组
result = np.vstack((array1, array2))
print(result)
示例1:
示例2:
import numpy as np
# 假设我们有三组一维数据
data1 = np.array([1, 2])
data2 = np.array([3, 4])
data3 = np.array([5, 6])
# 将一维数据转换为二维列向量
data1_2d = data1[:, np.newaxis]
data2_2d = data2[:, np.newaxis]
data3_2d = data3[:, np.newaxis]
# 使用np.vstack()函数将列向量堆叠起来
stacked_data = np.vstack((data1_2d, data2_2d, data3_2d))
print(stacked_data)
示例2:
注:使用np.vstack()进行堆叠时,输入数组的列数必须相同。如果列数不一致,将会引发一个ValueError异常。
此外,np.vstack()返回的是一个新的数组对象,原始数组不会被修改。