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

一、List

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

结果:2

二、 array多维矩阵中返回最大值下标

一维和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)

结果:

你可能感兴趣的:(python,python,numpy,机器学习)