Python如何将numpy一维数组转成三维数组?

1 个回答|283 次阅读
Coco老师 - 公众号:量位学堂Qbit编程
擅长:编程教育

将以下包含 12 个元素的一维数组转换为三维数组。

最外层维度将有 2 个数组,其中包含 3 个数组,每个数组有 2 个元素:

import numpy as np
arr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12])
newarr = arr.reshape(2, 3, 2)
print(newarr)

执行结果:

[[[ 1  2]
[ 3 4]
[ 5 6]]
[[ 7 8]
[ 9 10]
[11 12]]]