Существует гораздо более простой способ использования str.replace
: 'test'.replace('t','y').replace('e','a')
Однако, если вы хотите заменить все больше и больше букв, str.translate
будет более эффективным:
from string import maketrans
trans_from = "te"
trans_to = "ya"
trans_model = maketrans(trans_from, trans_to)
'test'.translate(trans_model)
Или, если вы хотите сохранить свой код, вы можете использовать elif
:
def translate(phrase):
translation = ""
for letter in phrase:
if letter in "t":
translation = translation + "y"
elif letter in "e":
translation = translation + "a"
else:
translation = translation + letter
return translation
print(translate(input("Enter word: ")))