将循环转换为列表理解

问题描述

嘿，所以我最近在将这个 for 循环转换为列表理解时遇到了麻烦

def str_translate_101(original,old,new):
result = ""
for letter in original:
    if letter == old:
        result += new
    else:
        result += letter
return result

有人可以帮忙吗？

解决方法

你需要一个推导式和一个三元运算符：

return [letter if letter == old else new for letter in original]

,

您可以做到，但它比普通的列表推导式要复杂一些，因为您要处理的是字符串而不是实际的 list。

def str_translate_101(original,old,new):
    result = "".join(new if letter == old else letter for letter in original)
    return result

这将返回一个字符串，与您的原始函数相同。