如果您不知道数组的形状,是否有一种简单的方法可以提取 ndarray 的第一项?
例如。给定以下数组:
arr = np.array([[[1,2,3,4], [5,6,7,8], [9,10,11,12]]])
>>> [[[ 1 2 3 4]
[ 5 6 7 8]
[ 9 10 11 12]]]
Run Code Online (Sandbox Code Playgroud)
我不想1
假设我知道这个数组的形状是 1*3*4。
我还对最小化解决方案的内存和 CPU 要求感兴趣。