NumPy的hstack函数详细教程

目录

• 1. 函数基本语法
• 2. 一维数组的堆叠
• 3. 二维数组的堆叠
• 4. 三维数组的堆叠
• 5. 注意事项
• 7. 与其他堆叠函数比较
• 8. 实际应用示例

`np.hstack()`是NumPy中用于水平（按列）堆叠数组的函数（这意味着它将数组在第二个轴（即列方向）上堆叠，但是要求除第二个轴外其他轴的大小必须相同）。下面通过详细的解释和示例来学习这个函数。

numpy.hstack(tup)

参数：
– `tup`：包含要堆叠数组的序列（通常是元组或列表），所有数组必须具有相同的形状（除了第二个轴，即列方向）

返回值：
– 堆叠后的数组

一维数组的水平堆叠会创建一个更长的一维数组：

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,)

二维数组的水平堆叠会增加列数：

# 二维数组示例
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)

对于三维数组，`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)

1. 形状要求：所有输入数组在除了第二个轴以外的所有轴上必须具有相同的形状
2. 错误示例：
 

   # 这会报错，因为行数不同
   a = np.array([[1, 2], [3, 4]])
   b = np.array([[5, 6, 7]])  # 形状不匹配
   # result = np.hstack((a, b))  # ValueError

– `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]]]

# 合并特征矩阵
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`函数的工作原理和适用场景。记住关键点是：**水平堆叠会增加数组的列数（第二个维度）**，并且所有输入数组在除了第二个维度外的其他维度上必须具有相同的形状。

到此这篇关于NumPy的hstack函数详细教程的文章就介绍到这了,更多相关NumPy hstack函数内容请搜索风君子博客以前的文章或继续浏览下面的相关文章希望大家以后多多支持风君子博客！

您可能感兴趣的文章:

• Numpy中vstack()和hstack()的使用方式
• Numpy中np.vstack() 和 np.hstack() 的实现
• numpy中hstack vstack stack concatenate函数示例详解
• numpy中数组拼接、数组合并方法总结(append(), concatenate, hstack, vstack, column_stack, row_stack, np.r_, np.c_等)
• numpy中np.dstack()、np.hstack()、np.vstack()用法
• numpy数组做图片拼接的实现（concatenate、vstack、hstack）
• Numpy中stack(),hstack(),vstack()函数用法介绍及实例
• Python numpy实现数组合并实例(vstack,hstack)