如何使用 numpy 添加矩阵的所有元素？

Question

I have a function called check_sum that adds all the elements of an 2d array.

Grid is a 2d array, and grid_shape is 1024x1024.

I'm trying to use the numpy.matrix.sum function, but after several attempts I don't know how to implement it, it always gives me another value.

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

You can call grid.sum() and it will return the sum of all the entries in your array, no matter if it is a 2d or a 3d array. Sum() is a general numpy method and it works both for matrices and arrays. here