|
`np.hstack()`是NumPy中用于水平（按列）堆叠数组的函数（这意味着它将数组在第二个轴（即列方向）上堆叠，但是要求除第二个轴外其他轴的大小必须相同）。下面通过详细的解释和示例来学习这个函数。
1. 函数基本语法
参数：
- `tup`：包含要堆叠数组的序列（通常是元组或列表），所有数组必须具有相同的形状（除了第二个轴，即列方向）
返回值：
- 堆叠后的数组
2. 一维数组的堆叠
一维数组的水平堆叠会创建一个更长的一维数组：
import numpy as np
# 一维数组示例
a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
print("a:", a)
print("b:", b)
print("a.shape:", a.shape)
print("b.shape:", b.shape)
result = np.hstack((a, b))
print("hstack result:", result)
print("result.shape:", result.shape)输出：
a: [1 2 3]
b: [4 5 6]
a.shape: (3,)
b.shape: (3,)
hstack result: [1 2 3 4 5 6]
result.shape: (6,)
3. 二维数组的堆叠
二维数组的水平堆叠会增加列数：
# 二维数组示例
a = np.array([[1, 2], [3, 4]])
b = np.array([[5, 6], [7, 8]])
print("a:")
print(a)
print("b:")
print(b)
print("a.shape:", a.shape)
print("b.shape:", b.shape)
result = np.hstack((a, b))
print("hstack result:")
print(result)
print("result.shape:", result.shape)输出：
a:
[[1 2]
[3 4]]
b:
[[5 6]
[7 8]]
a.shape: (2, 2)
b.shape: (2, 2)
hstack result:
[[1 2 5 6]
[3 4 7 8]]
result.shape: (2, 4)
4. 三维数组的堆叠
对于三维数组，`hstack`会在第二个维度（列）上堆叠：
# 三维数组示例
a = np.random.randn(2, 3, 4)
b = np.random.randn(2, 2, 4)
print("a.shape:", a.shape)
print("b.shape:", b.shape)
result = np.hstack((a, b))
print("result.shape:", result.shape)输出：
a.shape: (2, 3, 4)
b.shape: (2, 2, 4)
result.shape: (2, 5, 4)
5. 注意事项
1. 形状要求：所有输入数组在除了第二个轴以外的所有轴上必须具有相同的形状
2. 错误示例：
# 这会报错，因为行数不同
a = np.array([[1, 2], [3, 4]])
b = np.array([[5, 6, 7]]) # 形状不匹配
# result = np.hstack((a, b)) # ValueError
7. 与其他堆叠函数比较
- `vstack()`：垂直堆叠（按行）
- `dstack()`：深度堆叠（沿第三个轴）
- `concatenate()`：通用连接函数，可以指定轴```python
比较不同堆叠方法
a = np.array([[1, 2], [3, 4]])
b = np.array([[5, 6], [7, 8]])
print("hstack (水平):")
print(np.hstack((a, b)))
print("vstack (垂直):")
print(np.vstack((a, b)))
print("dstack (深度):")
print(np.dstack((a, b)))输出
hstack (水平):
[[1 2 5 6]
[3 4 7 8]]
vstack (垂直):
[[1 2]
[3 4]
[5 6]
[7 8]]
dstack (深度):
[[[1 5]
[2 6]]
[[3 7]
[4 8]]]
8. 实际应用示例
# 合并特征矩阵
features1 = np.random.randn(100, 5) # 100个样本，5个特征
features2 = np.random.randn(100, 3) # 100个样本，3个特征
combined_features = np.hstack((features1, features2))
print("原始特征形状:", features1.shape, features2.shape)
print("合并后特征形状:", combined_features.shape)通过这个教程，你应该能够理解`hstack`函数的工作原理和适用场景。记住关键点是：**水平堆叠会增加数组的列数（第二个维度）**，并且所有输入数组在除了第二个维度外的其他维度上必须具有相同的形状。 |
发表于 2026-1-6 10:56:59