论文复现 np.hstack的用法案例

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适合需要沿数组的高度（垂直方向）组合数据的情况。