python：返回列表和数组的最大最小值的索引

常常需要找到一个列表中或者数组中最小的值并返回索引，经常用，经常忘，害。

1、list中返回索引

list1 = [1,2,3,2,1]
print(list1.index(max(list1)))
#结果：2
list2 = [1,2,3,4]
print(list2.index(min(list2)))
#结果：0

其中返回最大值 直接使用max即可。

2、array中返回索引

1、直接用数组形式进行输入

>>> arr1 = np.array([1,2,3,4])
>>> np.max(arr1)
4
>>> np.argmax(arr1)
3

argmax()的输入直接是一个数组，直接返回最大值的索引，最小值索引argmin。

其中获得最大值：max(arr,axis = None)

>>> arr = np.arange(9).reshape((3,3))
>>> arr
array([[0, 1, 2],
       [3, 4, 5],
       [6, 7, 8]])
>>> np.max(arr)  # 获得全局最大值
8
>>> np.max(arr, axis = 1)  # 按行取得最大值
array([2, 5, 8])
>>> np.max(arr, axis = 0)  # 按列取得最大值
array([6, 7, 8])

然后用where也能得到最大值的索引，返回值中，前面的array对应行数，后者对应列数

>>> print(np.where(arr==np.max(arr)))
(array([2], dtype=int64), array([2], dtype=int64)) #表示最大值在第二行第二列
>>> print(np.where(arr==np.max(arr,axis=0)))
(array([2, 2, 2], dtype=int64), array([0, 1, 2], dtype=int64)) # 表示最大值分别在第二行第零列，第二行第一列，第二行第二列

如果array中有相同的最大值，where会将其位置全部给出

>>> arr[0,0] = 8
>>> arr[0,2] = 8
>>> arr
array([[8, 1, 8],
       [3, 4, 5],
       [6, 7, 8]])
>>> print(np.where (arr==np.max(arr)))
(array([0, 0, 2], dtype=int64), array([0, 2, 2], dtype=int64))

2、可以先将数组转换为list，再用list的方法进行获取。

>>> arr1 = np.array([1,2,3,4])
>>>list1 = arr1.tolist()
>>> np.argmax(arr1)
3
>>> list1.index(max(list1))
3

对数组进行编号在排序

又是需要通过两个数组其中一个排序通过序列索引对另一个数组进行排序
np.lexsort(A,B): 通过对B从小到大排序的索引对A在进行排序

>>> list2 = [3,4,2,1]
>>> arr2 = np.array((np.arange(len(list2)),list2)).T
>>> arr2
array([[0, 3],
       [1, 4],
       [2, 2],
       [3, 1]])

>>> list2
[3, 4, 2, 1]
>>> index = np.lexsort((arr2[:,0],arr2[:,1]))  # 通过对arr2[:1]J进行排序返回索引
>>> index 
array([3, 2, 0, 1], dtype=int64)
>>> arr2[index] # 完成对arr2[:1]进行排序，同时对应的arr[:,0]相应进行改变
array([[3, 1],
       [2, 2],
       [0, 3],
       [1, 4]])

参考文章

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