NumPy Nansum через первый индекс

Question

У меня есть пример массива 2 x 2 x 2:

np.array([[[ 1,  2],
        [ 3,  4]],

       [[ 5,  6],
        [ 7 ,  8]]])

Я хочу, чтобы nansum массива по первому индексу был следующим:

Суммировать все значения в:

[[ 1,  2],
[ 3,  4]]

и

[[ 5,  6],
[ 7 ,  8]]

Сумма первого массива будет равна 10, а второго - 26

, то есть

array([10, 26])

Answer 1

Я думаю, что вы ищете это

a = np.array([[[ 1,  2],
               [ 3,  4]],
             [[ 5,  6],
              [ 7 ,  8]]])
np.nansum(a,axis=(1,2))
# array([10, 26])

, потому что вы хотите суммировать только по осям 1 и 2 и получить одно число на ось 0