numpy 将一维数组和二维数组拼接

NumPy是一个常用的Python库，用于科学计算。通过它，我们可以快速地创建、操作和处理多维数组。其中一项常见的操作就是数组的拼接。这个操作可以将多个数组按照行或列的顺序拼接在一起，从而形成一个新的数组。

对于一维数组的拼接，NumPy提供了concatenate()函数。这个函数可以将多个一维数组按照指定的轴（默认是0）拼接在一起，形成一个新的数组。例如，假设我们有两个一维数组a和b：

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

我们可以使用concatenate()函数将它们拼接在一起：

c = np.concatenate([a, b])
print(c)

输出为：

[1 2 3 4 5 6]

对于二维数组的拼接，NumPy也提供了hstack()和vstack()函数。这两个函数可以将多个数组按照行或列的顺序拼接在一起，形成一个新的数组。

hstack()函数将多个数组按照行的方向拼接在一起，即水平拼接。例如，假设我们有两个二维数组a和b：

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

我们可以使用hstack()函数将它们水平拼接在一起：

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

输出为：

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

vstack()函数将多个数组按照行的方向拼接在一起，即垂直拼接。例如，假设我们有两个二维数组a和b：

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

我们可以使用vstack()函数将它们垂直拼接在一起：

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

输出为：

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

通过以上介绍，我们可以实现一维数组和二维数组之间的拼接。