下面是一个将2D数组转换为3D数组的示例代码:
import numpy as np
# 2D数组
arr_2d = np.array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
# 将2D数组转为3D数组
arr_3d = np.expand_dims(arr_2d, axis=0)
# 打印结果
print("2D数组:")
print(arr_2d)
print("3D数组:")
print(arr_3d)
输出结果:
2D数组:
[[1 2 3]
[4 5 6]
[7 8 9]]
3D数组:
[[[1 2 3]
[4 5 6]
[7 8 9]]]
在示例代码中,我们使用np.expand_dims
函数将2D数组arr_2d
在第0轴上扩展了一个维度,从而将其转换为3D数组arr_3d
。