Параметр axis
Концепция: какую ось «сворачивает» агрегирующая функция.
Сигнатура
func(a, axis=...)axis указывает, по какой оси выполнять операцию и какая ось исчезнет из результата. Для матрицы (строки, столбцы): axis=0 — вдоль строк (по столбцам, результат — строка), axis=1 — вдоль столбцов (по строкам, результат — столбец).
axis=0: «сжимаем» вертикальноaxis=1: «сжимаем» горизонтально
import numpy as np
a = np.array([[1, 2, 3], [4, 5, 6]])
print(a.sum(axis=0)) # [5 7 9]
print(a.sum(axis=1)) # [ 6 15]