Avatar ·

A melhor forma de trocar as letras "A" e "B" de lugar

📁 função, python, exemplo

Existe uma função que recebe um argumento na forma de uma string composta por "a", "A", "b" e "B".

Cada "b" deve ser substituído por "a", cada "a" por "B", e também com o registro minúsculo.

>>> swap('aabbbaa')
'BBAAAAABB'

Escrevi um código fraco, mas funcional:

def swap(text):
    new_text = []
    for c in text:
        if c == 'a':
            new_text.append('B')
        elif c == 'b':
            new_text.append('A')
        elif c == 'A':
            new_text.append('b')
        elif c == 'B':
            new_text.append('a')
    return ''.join(new_text)