[英]How to cut a Numpy 2d array and store as a 3d array?
假设我有以下 Numpy 二维数组 (3*9)
[[-2 -2 -2 -2 -2 -2 -2 -2 -2]
[-2 -2 -2 -2 -2 -2 -2 -2 -2]
[-2 -2 71 -1 -1 -1 71 -1 52]]
我想将整个数组除以 3 并得到这样的目标数组,这意味着我应该得到一个形状为 3d 的数组(3 x 3 x3):
[[-2 -2 -2] [-2 -2 -2] [-2 -2 -2]
[-2 -2 -2] [-2 -2 -2] [-2 -2 -2]
[-2 -2 71] [-1 -1 -1] [71 -1 52]]
任何人都可以解释如何做到这一点?
如果你已经知道之前的形状和目标形状,你可以简单地使用reshape
arr = [[-2, -2, -2, -2, -2, -2, -2, -2, -2],
[-2, -2, -2, -2, -2, -2, -2, -2, -2,],
[-2, -2, 71, -1, -1, -1, 71, -1, 52]]
arr = np.array(arr)
arr.reshape(3,3,3)
您可以使用如下reshape
方法
import numpy as np
array = np.array([[-2,-2,-2,-2,-2,-2,-2,-2,-2],
[-2,-2,-2,-2,-2,-2,-2,-2,-2],
[-2,-2,71,-1,-1,-1,71,-1,52]])
newarr = array.reshape(3,3,3)
print(newarr)
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.