获取多维NumPy数组中最大值的位置
获取多维NumPy数组中最大值的位置
提问于 2010-08-27 20:46:53
如何获取多维NumPy数组中最大值的位置(索引)？
回答 2
Stack Overflow用户
发布于 2019-01-13 06:11:43
您可以简单地编写一个函数(仅适用于2d)：
def argmax_2d(matrix):
maxN = np.argmax(matrix)
(xD,yD) = matrix.shape
if maxN >= xD:
x = maxN//xD
y = maxN % xD
else:
y = maxN
x = 0
return (x,y)Stack Overflow用户
发布于 2020-09-05 04:21:41
另一种方法是将numpy数组更改为list，并使用max和index方法：
List = np.array([34, 7, 33, 10, 89, 22, -5])
_max = List.tolist().index(max(List))
_max
>>> 4页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/3584243
复制相关文章
相似问题
腾讯云开发者
