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表示数组的第三维。
原文地址: http://www.cveoy.top/t/topic/bs9q 著作权归作者所有。请勿转载和采集!