Статистика по осям

Два кафе подсчитывали количество проданных чашек кофе в течение трёх дней: одна строка на кафе, один столбец на день. Ось задаёт направление, в котором нужно складывать:

Вы передаёте его по имени:

cups = np.array([[30, 45, 24],
                 [50, 40, 36]])
print(cups.sum(axis=0))
print(cups.sum(axis=1))

Это выводит:

[80 85 60]
[ 99 126]

Ось, которую ты указываешь, исчезает: axis=0 сворачивает строки, оставляя один результат на столбец. В данных ML это одно число на признак.

.argmax() показывает, где находится наибольшее значение: cups.sum(axis=0).argmax() равно 1, то есть вторник.