可以使用 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 表示数组的第三维。

NumPy 二维数组转换为三维数组:完整指南

原文地址: https://www.cveoy.top/t/topic/mWBM 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录