Функция вдоль оси в NumPy

Функция apply_along_axis берёт таблицу, выбирает ось и для каждого среза вдоль неё вызывает вашу функцию: на вход подаётся одномерный фрагмент, на выходе собирается новый массив.

Посчитаем сумму баллов в каждой строке таблицы:

import numpy as np

def row_sum(row):
	return np.sum(row)

scores = np.array([[1, 2, 3],
                   [4, 0, 1]])
totals = np.apply_along_axis(row_sum, 1, scores)
print(totals) # выведет [6 5]

Для таблицы цен [[10, 5], [2, 3]] получите сумму по каждой строке с помощью своей функции над одномерным рядом и выведите ответ.

Для оценок [[4, 4], [1, 2], [3, 3]] найдите наибольший балл в каждой строке и выведите ряд результатов.