如何连接两个二维 NumPy 数组

在进行 NumPy 编程时，我们有时需要将两个二维数组连接起来。NumPy 库提供了很多方法来连接数组。本文将介绍一些常见的方法。

水平连接（hstack）

import numpy as np

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

c = np.hstack((a, b))
print(c)

输出：

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

以上代码使用了 hstack 函数来连接数组。该函数将两个数组水平堆叠在一起。

垂直连接（vstack）

import numpy as np

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

c = np.vstack((a, b))
print(c)

输出：

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

以上代码使用了 vstack 函数来连接数组。该函数将两个数组垂直堆叠在一起。

深度连接（dstack）

import numpy as np

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

c = np.dstack((a, b))
print(c)

输出：

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

以上代码使用了 dstack 函数来连接数组。该函数将两个数组深度堆叠在一起。

列连接（column_stack）

import numpy as np

a = np.array([1, 2, 3])
b = np.array([4, 5, 6])

c = np.column_stack((a, b))
print(c)

输出：

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

以上代码使用了 column_stack 函数来连接数组。该函数将两个一维数组按列连接成一个二维数组。

行连接（row_stack）

import numpy as np

a = np.array([1, 2, 3])
b = np.array([4, 5, 6])

c = np.row_stack((a, b))
print(c)

输出：

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

以上代码使用了 row_stack 函数来连接数组。该函数将两个一维数组按行连接成一个二维数组。

总结

以上介绍了 NumPy 库中常用的数组连接方法。使用这些方法可以快速、方便地处理二维数组的连接问题。