python 如何获得list或numpy数组中最大元素对应的索引

目录


前言

例如:随着人工智能的不断发展,机器学习这门技术也越来越重要,很多人都开启了学习机器学习,本文就介绍了机器学习的基础内容。

一:list

1.1获得list中最大元素的索引

a = [1,2,3,4,5]
a.index(max(a))

python 如何获得list或numpy数组中最大元素对应的索引

相应的最小值使用

a = [1,2,3,4,5]
a.index(min(a))

python 如何获得list或numpy数组中最大元素对应的索引

 

二 numpy数组中

获得numpy数组中最大元素的索引

2.1 可以使用numpy的函数,argmax获得最大元素的索引,相应的获得最小值的话需要使用argmin

arr = np.array([1,2,3,4,5])
maxindex = np.argmax(arr )
maxindex

 python 如何获得list或numpy数组中最大元素对应的索引

 

 2.2 也可以将numpy转为list,然后使用list或者最大值索引的方法获得最大值。

a = np.array([1,2,3,4,5])
b = a.tolist()
b.index(max(b))

python 如何获得list或numpy数组中最大元素对应的索引

 

三:补:where 函数

3.1 np.max函数

3.2 np.where函数

numpy.where(condition[,x,y])

3.3 实例:

a = np.arange(9).reshape((3,3))
print(a)
print(type(array))
print(np.max(a))    #全局最大
print(np.max(a,axis=0)) #axis 按列查找  找每一列元素最大值 |3列所以结果为列表大小为3
print(np.max(a,axis=1)) #按行查找  找每一行元素最大值

print(np.where(a==np.max(a)))
print(np.where(a==np.max(a,axis=0)))

python 如何获得list或numpy数组中最大元素对应的索引

 

总结

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
心中带点小风骚的头像心中带点小风骚普通用户
上一篇 2023年3月10日
下一篇 2023年3月10日

相关推荐