如何替换字符串python中的正确字母

Question

作业是:

您的任务是更正数字化文本中的错误.您只需要处理以下错误:

• S被误解为5
• O被误解为0
• 我被误解为1

我的代码:

def correct(string):
    for i in string:
        if '5' in string:
           string = string.replace('5','S') 
        elif '0' in string:
          string = string.replace('0','O')
        elif '1' in string:
            string = string.replace('1','I')
    return string 

我知道这个解决方案不适用于:

Test.assert_equals(correct("51NGAP0RE"),"SINGAPORE");

有没有人有关于如何使这个更通用的功能,适用于每个单词的提示？

Answer 1

你可以str.replace直接使用.

def correct(string):
    return string.replace('5','S').replace('0','O').replace('1','I')

Answer 2

你为什么不利用str.maketrans和str.translate:

>>> "51NGAP0RE".translate(str.maketrans('501', 'SOI'))
'SINGAPORE'

包含在一个功能中:

def correct(s):
    return s.translate(str.maketrans('501', 'SOI'))