import numpy as np

## arrInput = [5, 2, 9, 1, 7, 6]  ## Example 1
arrInput = [4, 2, 3, 1, 5, 0]  ## Example 2

print (np.argmax(arrInput))