【Python】np.hstack()和np.vstack函数详解和示例

本文通过函数原理和运行示例，对np.hstack()和np.vstack函数进行详解，以帮助大家理解和使用。
更多Numpy函数详解和示例，可参考

【Python】Numpy库近50个常用函数详解和示例，可作为工具手册使用

np.hstack()

np.hstack() 是 NumPy 库中的一个函数，用于将两个或更多的数组沿着水平轴（即列）连接起来。这个函数会沿着水平轴（即列）将输入的数组堆叠起来。这意味着输出的数组的列数将是输入数组的列数之和，而行数将是输入数组中最大行数的值。

函数解析

函数原型为：
numpy.hstack(tup)
其中 tup 是一个元组，包含了你想要堆叠的数组。

运行示例

一维数组

import numpy as np

# 创建两个一维数组
a = np.array([1, 2, 3])
b = np.array([4, 5, 6])

# 使用 np.hstack() 将它们堆叠起来
c = np.hstack((a, b))
print(c)  

输出：

[1 2 3 4 5 6]

二维数组

import numpy as np

# 创建两个二维数组
a = np.array([[1, 2], [3, 4]])
b = np.array([[5, 6], [7, 8]])

# 使用 np.hstack() 将它们堆叠起来
c = np.hstack((a, b))

print(c)

输出：

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

np.vstack()

在NumPy中，用于按行堆叠（即在水平方向上堆叠）的函数是np.vstack()。这个函数将两个或更多的数组沿着垂直轴（即行）堆叠起来。

函数解析

数原型为：
numpy.vstack(tup)
其中tup是一个元组，包含了你想要堆叠的数组。

运行示例

import numpy as np

# 创建两个二维数组
a = np.array([[1, 2], [3, 4]])
b = np.array([[5, 6], [7, 8]])
# 使用 np.vstack() 将它们堆叠起来
c = np.vstack((a, b))
print("按行堆叠:")
print(c)

输出：

按行堆叠:
[[1 2]
 [3 4]
 [5 6]
 [7 8]]