numpy如何获取多维矩阵中最大值的坐标

史丹利复合田

于 2019-07-28 08:15:50 发布

阅读量1.5w

点赞数 11

CC 4.0 BY-SA版权

分类专栏： python

本文链接：https://blog.youkuaiyun.com/tsyccnh/article/details/97610371

7 篇文章

订阅专栏

本文介绍了一种使用Python的numpy库中argmax和unravel_index函数来确定多维数组中最大值位置的方法。这种方法适用于任何维度的矩阵，通过实例演示了如何找到并打印出最大元素的位置。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

利用argmax和unravel_index函数即可。任何维度的矩阵都可以。

import numpy as np
a = np.array([[1,2,3],[4,5,6],[3,2,1]])
pos = np.unravel_index(np.argmax(a),a.shape)
print(pos)
---------------
(1, 2)