which.max python code example

Example 1: python how to return max num index

import numpy as np

x=[1,2,3,4,5]

max_index=np.argmax(x) 
# return the index of the maximun element
# max_index has the value 4

Example 2: index of maximum value in list python

[i for i, x in enumerate(l) if x == max(l)]