get index of max value in list python numpy