Python 将一维数组或矩阵变为三维

正文

话不多说直接上代码：

import numpy as np

sampling_points = 10001

arr = np.linspace(0, 2, sampling_points)
arr_3D = arr.reshape(1, 1, -1)
print(arr_3D)
"""
result:
[[[0.0000e+00 2.0000e-04 4.0000e-04 ... 1.9996e+00 1.9998e+00 2.0000e+00]]]
"""

可以看到，此时我们得到的三维数组，或者说矩阵所有的元素都是沿着 x 轴排列的。当然，也可以将 reshape 中的参数更改为以下形式：

# 元素沿着 y 轴排列
arr_3D = arr.reshape(1, -1, 1)
# 元素沿着 z 轴排列
arr_3D = arr.reshape(-1, 1, 1)

关于为什么是这样，可以参考 numpy数组的坐标轴问题。

如果大家觉得有用，就请点个赞吧~