Как я могу добавить все элементы матрицы, используя NumPy?

Question

У меня есть функция с именем check_sum, которая добавляет все элементы двухмерного массива.

Сетка - это двумерный массив, а grid_shape - 1024x1024.

Я пытаюсь использовать функцию numpy.matrix.sum, но после нескольких попыток не знаю, как ее реализовать, это всегда дает мне другое значение.

def check_sum(grid):
  xmax, ymax = grid_shape
  s=0
  for i in range(xmax):
    for j in range(ymax):
      s += grid[i][j]
  return s

Answer 1

Вы можете вызвать grid.sum(), и он вернет сумму всех записей в вашем массиве, независимо от того, является ли он двумерным или трехмерным массивом. Sum() - это общий метод NumPy, и он работает как для матриц, так и для массивов. здесь