Python基础操作——list返回最大值下标、array多维返回最大值下标

泡泡怡

于 2022-09-11 18:34:28 发布

阅读量3.1k

点赞数 4

CC 4.0 BY-SA版权

分类专栏： python 文章标签： python numpy 机器学习

本文链接：https://blog.youkuaiyun.com/m0_72662900/article/details/126807760

39 篇文章

订阅专栏

本文介绍了如何使用Python的NumPy库在多维数组中找到最大值及其对应的索引，通过实例展示了如何应用`np.argmax`函数并获取unravel_index结果。

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

一、List

list1=list([1,2,3])
list1
k=list1.index(max(list1))
k

结果：2

一维和list一样。

import numpy as np
A=np.array([[0,1,1],[0,4,-1],[2,-2,1]])
max_index = np.unravel_index(np.argmax(A, axis=None), A.shape)
print(max_index)

结果：