python合并不同维度的数组

在 Python 中，可以使用 NumPy 库中的 concatenate 函数来合并不同维度的数组。以下是一个示例代码，可以通过该代码来合并两个不同维度的数组：

import numpy as np

# 创建一个二维数组
arr1 = np.array([[1, 2], [3, 4]])

# 创建一个一维数组
arr2 = np.array([5, 6])

# 将 arr2 添加到 arr1 的第一维末尾
result = np.concatenate((arr1, np.expand_dims(arr2, axis=0)), axis=0)

print(result)

在上述代码中，首先创建一个二维数组 arr1 和一个一维数组 arr2。然后，使用 concatenate 函数将 arr2 添加到 arr1 的第一维末尾，并将结果保存在变量 result 中。在调用 concatenate 函数时，需要将 arr2 使用 np.expand_dims 函数扩展为一个二维数组，这样才能与 arr1 进行合并。最终输出合并后的结果。

需要注意的是，使用 concatenate 函数合并数组时，需要保证除合并维度外的其他维度的大小相同，否则会出现 ValueError 错误。此外，还可以使用 vstack 函数和 hstack 函数来分别对两个数组进行垂直方向和水平方向的合并。