python把两个np.array拼接

可以使用 numpy 库中的 concatenate 函数来拼接两个 numpy 数组。

例如，假设有两个数组 a 和 b，可以使用以下代码将它们沿着某个轴拼接起来：

import numpy as np

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

# 沿着第 0 轴拼接
c = np.concatenate((a, b), axis=0)
print(c)
# 输出:
# [[1 2]
#  [3 4]
#  [5 6]]

其中，第一个参数传入要拼接的数组序列，第二个参数 axis 指定沿着哪个轴进行拼接。在上面的例子中，沿着第 0 轴（行）进行拼接。

如果要在水平方向拼接，则可以将 axis 参数设置为 1。

import numpy as np

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

# 沿着第 1 轴拼接
c = np.concatenate((a, b), axis=1)
print(c)
# 输出:
# [[1 2 5]
#  [3 4 6]]