numpy获取多维矩阵的最大值

需要求一个41610243矩阵的最大值,百度了一下发现都是求二维矩阵的,因此,获取多维矩阵的最大值有以下两种方法:

  1. 把矩阵分成三个二维矩阵,分别得到最大值,然后再得到三个最大值中的最大值
  2. 曲线救国。百度的时候发现这位大佬写了一篇求多维矩阵最大值坐标的文章
    pos = np.unravel_index(np.argmax(a),a.shape)
    其中a为任意维度矩阵。
    得到最大值坐标后,多维矩阵的最大值也就很方便地可以得到啦

你可能感兴趣的:(python)