numpy hstack

numpy hstack

在numpy中，hstack函数用于沿着水平方向堆叠数组（即按列连接数组）。这个函数可以用来将多个数组按列连接起来，生成一个新的数组。

语法

numpy.hstack(tup)

tup是一个包含numpy数组的元组或列表。

参数

• tup：包含numpy数组的元组或列表。

返回值

返回一个水平堆叠后的numpy数组。

示例

import numpy as np

array1 = np.array([[1, 2], [3, 4]])
array2 = np.array([[5, 6], [7, 8]])

result = np.hstack((array1, array2))
print(result)

运行结果

array([[1, 2, 5, 6],
       [3, 4, 7, 8]])

通过以上示例代码可以看到，hstack函数将array1和array2沿着水平方向堆叠在一起，生成了一个新的数组，新数组的维度是2×4。

在实际应用中，hstack函数非常常用，特别是在处理多个不同数组时，将它们水平堆叠在一起可以方便后续的数据处理。另外，如果想要垂直堆叠数组，可以使用vstack函数。