如果您不知道数组的形状,是否有一种简单的方法可以提取 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]]]
我不想1假设我知道这个数组的形状是 1*3*4。
我还对最小化解决方案的内存和 CPU 要求感兴趣。