Злой код из библиотеки стандарта Python

Вы можете изменить свой код, как показано ниже:

def scrambleWord(userInput):
    count = 0
    scramble = ''
    while count < len(userInput):
        if count < len(userInput)-1 and userInput[count] =='A' and userInput[count+1] != 'A':
            scramble+= userInput[count+1] + userInput[count] 
            count+=2
        else:
            scramble += userInput[count]
            count+=1
    return scramble

Вы не проверяете условие (count < len(userInput)-1), когда логика пытается проверить наличие A и поменять местами со следующей буквой. Выдает строковый индекс вне диапазона исключений.