np.hstack
是NumPy库中的一个函数,用于水平堆叠(即沿着水平轴)两个或多个数组。这个函数对于组合数据或对齐列非常有用。以下是np.hstack
的一些使用示例。
基本用法
假设我们有两个一维数组,我们想要将它们水平堆叠到一起:
import numpy as np
# 创建两个一维数组
a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
# 水平堆叠
result = np.hstack((a, b))
print(result)
这会输出:
[1 2 3 4 5 6]
二维数组的用法
np.hstack
也可以用来处理二维数组(即矩阵)。当处理二维数组时,它会将数组沿着第二个轴(即列)堆叠起来。
import numpy as np
# 创建两个二维数组
a = np.array([[1, 2], [3, 4]])
b = np.array([[5, 6], [7, 8]])
# 水平堆叠
result = np.hstack((a, b))
print(result)
这会输出:
[[1 2 5 6]
[3 4 7 8]]
与其他堆叠函数的对比
对比其他堆叠函数,比如np.vstack
(垂直堆叠):
# 创建两个一维数组
a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
# 垂直堆叠
result_vstack = np.vstack((a, b))
print("Vertical stack:\n", result_vstack)
垂直堆叠的输出为:
Vertical stack:
[[1 2 3]
[4 5 6]]
np.hstack
适合需要沿数组的宽度(水平方向)组合数据的情况,而np.vstack
适合需要沿数组的高度(垂直方向)组合数据的情况。