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

Question

我有一个名为 check_sum 的 function ，它添加了二维数组的所有元素。

Grid 是一个二维数组，grid_shape 是 1024x1024。

我正在尝试使用 numpy.matrix.sum function，但经过几次尝试我不知道如何实现它，它总是给我另一个价值。

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() ，它将返回数组中所有条目的总和，无论它是 2d 数组还是 3d 数组。 Sum()是一种通用的 numpy 方法，它适用于矩阵和 arrays。 这里