sum(axis=...)

Пример агрегации по конкретной оси на матрице.

Сигнатураa.sum(axis=0) / a.sum(axis=1)

Наглядный пример различия осей. axis=0 даёт суммы по столбцам, axis=1 — по строкам.

import numpy as np
a = np.array([[1, 2], [3, 4], [5, 6]])
print(a.sum(axis=0))   # [ 9 12]
print(a.sum(axis=1))   # [ 3  7 11]