Суммируйте одновременно 3 измерения 4-мерной матрицы

Question

у меня 4d матрица, mat4.Вместо использования np.sum(mat, axis=) и определения axis 3 раза для суммы трех измерений, есть ли способ суммировать одновременно много измерений матрицы?

#Sum `mat4` except `axis=0`

mat4 = np.random.rand(2,3,4,5)
matsum = np.sum(mat4, axis=3)
matsum = np.sum(matsum, axis=2)
matsum = np.sum(matsum, axis=1)

print matsum.shape
>> (2L,)

Answer 1

Ключевое слово

axis может быть либо int, либо tuple, поэтому вы можете просто использовать

np.sum(mat, axis=(1, 2, 3))

С np.sum документы:

Если ось является кортежем из целых чисел, сумма выполняется по всем осям, указанным в кортеже, а не по одной оси или по всем осям, как раньше.