Сохранение осей в NumPy

Без дополнительных параметров sum убирает ось, вдоль которой шло сложение: из таблицы получается ряд. Параметр keepdims со значением True оставляет эту ось длиной 1, и форма результата остаётся двумерной.

Сложим таблицу по столбцам с сохранением осей и выведем форму:

import numpy as np

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

Если сводка идёт по строкам, сохранённая ось стоит второй в форме:

import numpy as np

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

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

Для той же таблицы баллов сложите по строкам с сохранением осей и выведите форму результата.

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