NumPy vstack()

vstack() 方法将给定的输入数组序列垂直堆叠起来。

示例

import numpy as np

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

# stack the arrays stackedArray = np.vstack((array1, array2))
print(stackedArray) ''' Output [[0 1] [2 3] [4 5] [6 7]] '''

vstack() 语法

vstack() 的语法是

numpy.vstack(tup)

vstack() 参数

vstack() 方法接受一个参数

  • tup - 要堆叠的数组的元组

注意:给定元组中所有数组的形状必须相同,除了第一个维度,因为我们是在轴 0 上进行堆叠。


vstack() 返回值

vstack() 方法返回垂直堆叠后的数组。


示例 1:垂直堆叠数组

import numpy as np

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

# stack the arrays stackedArray = np.vstack((array1, array2, array3))
print(stackedArray)

输出

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

示例 2:垂直堆叠无效形状的数组

import numpy as np

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

# stacks the arrays 
stackedArray = np.vstack((array1, array2))

print(stackedArray)

输出

ValueError: all the input array dimensions except for the concatenation axis must match exactly, but along dimension 1, the array at index 0 has size 2 and the array at index 1 has size 3

我们的高级学习平台,凭借十多年的经验和数千条反馈创建。

以前所未有的方式学习和提高您的编程技能。

试用 Programiz PRO
  • 交互式课程
  • 证书
  • AI 帮助
  • 2000+ 挑战