Как я могу изменить свой код, чтобы изменить некоторые подстроки?

Question

Я пытаюсь переместить первые 2 символа строки до конца, а последние 2 символа - вперед. Затем я хочу поменять местами перемещенные символы.

Например, bajui. Сначала мы меняем первые и последние два символа, давая uijba, а затем я хочу обратить ui в iu и ba в ab, и, таким образом, конечный результат должен быть iujab.

Вот мой текущий код:

def startToEnd(kata):
   kata = kata[-2:] + kata[2:3] + kata[:2]
   return kata
print(startToEnd("bajui"))

с этим он даст uijba. Как я могу изменить символы, которые были перемещены?

Answer 1

Вы можете просто использовать нарезку строк для этого - обратите внимание, что это будет работать только так, как предназначено для строк как минимум длиной 4.

Код:

def startToEnd(kata):
    return kata[:-3:-1] + kata[2:-2] + kata[1::-1]

Выход:

>>> startToEnd('makam')
'makam'
>>> startToEnd("bajui")
'iujab'