NumPy 二维数组转换为三维数组:完整指南
可以使用 NumPy 的 reshape 函数将二维数组转换为三维数组。
例如,假设我们有一个二维数组 'arr',其形状为 (4, 6),我们想将其转换为一个三维数组,其形状为 (2, 3, 4)。可以使用以下代码:
import numpy as np
arr = np.arange(24).reshape(4, 6)
arr_3d = arr.reshape(2, 3, 4)
print(arr)
print(arr_3d)
输出:
[[ 0 1 2 3 4 5]
[ 6 7 8 9 10 11]
[12 13 14 15 16 17]
[18 19 20 21 22 23]]
[[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]]
[[12 13 14 15]
[16 17 18 19]
[20 21 22 23]]]
可以看到,原始的二维数组 'arr' 被成功转换为一个三维数组 'arr_3d'。在这个例子中,我们使用了 reshape 函数将原始数组的形状从 (4, 6) 变为 (2, 3, 4),其中 2 表示数组的第一维,3 表示数组的第二维,4 表示数组的第三维。
原文地址: https://www.cveoy.top/t/topic/mWBM 著作权归作者所有。请勿转载和采集!