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

任务是：

您的任务是纠正数字化文本中的错误。您只需处理以下错误：

• S 被误解为 5
• O 被误解为 0
• I 被误解为 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");

有谁知道如何使其成为适用于每个单词的更通用的功能吗？

您可以使用str.replace直接地。

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