可以使用numpy中的reshape
函数
将二维数组转换为三维数组,再将这两个三维数组沿着最后一个轴进行合并。
示例代码:
import numpy as np
# 创建二维数组
a = np.array([[1, 2], [3, 4]])
# 将二维数组转换为三维数组
a_3d = a.reshape((a.shape[0], a.shape[1], 1))
# 创建三维数组
b = np.array([[[5], [6]], [[7], [8]]])
# 沿着最后一个轴进行合并
c = np.concatenate((a_3d, b), axis=-1)
print(c)
# 输出结果:
# [[[1 5]
# [2 6]]
# [[3 7]
# [4 8]]]