📅  最后修改于: 2023-12-03 15:05:52.368000             🧑  作者: Mango
在NumPy中,vstack()函数用于沿着垂直方向将数组堆叠在一起。它可以帮助程序员在不改变任何数组形状的情况下,将它们连接在一起。下面是该函数的语法:
numpy.vstack(tup)
其中,参数tup是要堆叠的数组组成的序列,通常为元组。这些数组必须有相同的形状。如果数组的形状不同,则会抛出ValueError。
让我们看一些示例:
import numpy as np
# Example 1
a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
c = np.vstack((a, b))
print(c)
# Output: [[1 2 3]
# [4 5 6]]
# Example 2
x = np.array([[1], [2], [3]])
y = np.array([[4], [5], [6]])
z = np.vstack((x, y))
print(z)
# Output: [[1]
# [2]
# [3]
# [4]
# [5]
# [6]]
在上面的示例中,我们使用了vstack()函数将多个数组垂直堆叠在一起。
需要注意的是,vstack()函数只能用于numpy数组,而不能用于Python列表。所以,在堆叠列表时,需要先将列表转换为numpy数组,然后再使用vstack()函数。
vstack()函数还可以与其他NumPy函数一起使用,如np.split()和np.hsplit(),以便在数组中执行更多的操作。它是一个十分有用的函数,每个开发者在NumPy编程中都应该掌握。