numpy中把两个数组拼接的方法

在Python的NumPy库中，有多种方式可以把两个数组融合（合并）在一起。下面为你介绍几种常见的方法：

• numpy.concatenate(): 这个函数可以沿指定的轴把两个数组连接在一起。

import numpy as np

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

c = np.concatenate((a, b))
print(c)  # 输出：[1 2 3 4 5 6]

• numpy.vstack() 和 numpy.hstack(): vstack() 和 hstack() 函数分别用于垂直和水平方向的数组合并。

import numpy as np

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

# 针对1D数组，vstack相当于在列方向合并，形成2D数组
c = np.vstack((a, b))
print(c)
# 输出：
# [[1 2 3]
#  [4 5 6]]

# hstack直接在行方向合并，这对1D数组来说与concatenate效果相同
d = np.hstack((a, b))
print(d)  # 输出：[1 2 3 4 5 6]

• numpy.append(): 这个函数在一个数组的末尾添加元素。

import numpy as np

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

c = np.append(a, b)
print(c)  # 输出：[1 2 3 4 5 6]