Сумма формулой в NumPy

Функция einsum принимает строку индексов и один или несколько массивов. Если справа от стрелки нет ни одной буквы, в результате остаётся одно число: все оси, перечисленные слева, схлопываются суммированием.

Для одномерного ряда достаточно одной буквы оси и пустого ответа за стрелкой, запись 'i->' означает сумму по всем элементам ряда:

import numpy as np

arr = np.array([1, 2, 3])
total = np.einsum('i->', arr)
print(total) # выведет 6

У таблицы две буквы осей, запись 'ij->' суммирует и строки, и столбцы сразу, пока не останется один итог:

import numpy as np

table = np.array([[1, 2],
                  [3, 4]])
total = np.einsum('ij->', table)
print(total) # выведет 10

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

Для таблицы [[2, 0], [1, 3]] сложите все ячейки одной формулой с двумя буквами осей и выведите одно итоговое число.