【Python】np.hstack()和np.vstack函数详解和示例
np.hstack() 和 np.vstack() 是 NumPy 库中的两个常用函数，用于处理数组的堆叠。
np.hstack()：水平堆叠，即沿着水平轴（从左到右）堆叠数组。np.vstack()：垂直堆叠，即沿着垂直轴（从上到下）堆叠数组。
示例代码：
import numpy as np
# 创建两个形状相同的数组
array1 = np.array([1, 2, 3])
array2 = np.array([4, 5, 6])
# 水平堆叠
horizontal_stacked = np.hstack((array1, array2))
print(horizontal_stacked) # 输出: [1 2 3 4 5 6]
# 创建两个形状相同的数组
array3 = np.array([[1, 2, 3]])
array4 = np.array([[4, 5, 6]])
# 垂直堆叠
vertical_stacked = np.vstack((array3, array4))
print(vertical_stacked) # 输出: [[1 2 3] [4 5 6]]在这个例子中，array1 和 array2 水平堆叠得到一个新的一维数组，array3 和 array4 垂直堆叠得到一个二维数组。
评论已关闭