Что означает axis=1

В NumPy параметр axis=1 означает, что операция выполняется вдоль второй оси массива.

Для 2D массива (m, n):

• Первая ось (axis=0) — строки
• Вторая ось (axis=1) — столбцы

Соответственно:

• axis=1 → «собираем» элементы по горизонтали, вдоль столбцов, оставляя строки.

Пример:

import numpy as np
 
M = np.array([[1, 2, 3],
              [4, 5, 6]])
 
np.sum(M, axis=1)  # [6 15]

Объяснение:

• Берём каждую строку:

  • 1-я строка: 1 + 2 + 3 = 6
  • 2-я строка: 4 + 5 + 6 = 15

• Результат — массив с суммами по строкам, форма (m,)

Для nD массива (2, 3, 4):

• axis=1 → операция идёт вдоль второй оси (по «строкам внутри блоков»), оставляя первую и третью оси.

Коротко: axis=1 = по горизонтали, вдоль столбцов, оставляем строки.