如何使用 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。 這里