Сумма по оси в NumPy

У функции sum можно указать axis: тогда сводка идёт не по всем ячейкам, а вдоль выбранной линии. Ось с номером 0 идёт по столбцам, сверху вниз. Ось с номером 1 идёт по строкам, слева направо.

Возьмём таблицу из двух строк и трёх столбцов. Сложим по столбцам и выведем результат вместе с его формой:

import numpy as np

table = np.array([[1, 2, 3], [4, 5, 6]])
by_cols = np.sum(table, axis=0)
print(by_cols) # выведет [5 7 9]
print(np.shape(by_cols)) # выведет (3,)

Теперь сложим по строкам и снова посмотрим форму:

import numpy as np

table = np.array([[1, 2, 3], [4, 5, 6]])
by_rows = np.sum(table, axis=1)
print(by_rows) # выведет [ 6 15]
print(np.shape(by_rows)) # выведет (2,)

Для таблицы цен [[10, 20], [30, 40]] сложите по столбцам и выведите ряд сумм и его форму.

Для таблицы баллов [[1, 2, 3], [4, 5, 6]] сложите по строкам и выведите ряд сумм и его форму.

Для таблицы отметок [[2, 0], [1, 3], [4, 1]] сложите по столбцам и выведите ряд сумм.