相关疑难解决方法(0)

我想选择数组的某些元素,并根据值执行加权平均计算.但是,使用过滤条件会破坏数组的原始结构.arr它的形状(2, 2, 3, 2)变成了一维的阵列.这对我来说毫无用处,因为并非所有这些元素都需要在以后相互组合(但是它们的子阵列).我怎样才能避免这种扁平化呢？

>>> arr = np.asarray([ [[[1, 11], [2, 22], [3, 33]], [[4, 44], [5, 55], [6, 66]]], [ [[7, 77], [8, 88], [9, 99]], [[0, 32], [1, 33], [2, 34] ]] ])
>>> arr
array([[[[ 1, 11],
         [ 2, 22],
         [ 3, 33]],

        [[ 4, 44],
         [ 5, 55],
         [ 6, 66]]],


       [[[ 7, 77],
         [ 8, 88],
         [ 9, 99]],

        [[ 0, 32],
         [ 1, 33],
         [ 2, 34]]]])
>>> arr.shape
(2, …