以下是修改现有代码的方法,以便它能够满足您的需求.我已将newWord更改为new_word以符合标准的Python样式约定.
def replace(char, word1, word2):
new_word = ""
for i in range(len(word2)):
if word1[i] == char:
new_word += char
else:
new_word += word2[i]
return new_word
# Test
print(replace("p", "apple", "12345"))
产量
1pp45
请注意,如果word1比word2短,则此代码将引发IndexError:string索引超出范围异常.
有各种更好的方法来实现此功能.首先,我们可以使用zip函数并行迭代两个字符串的字符.这比使用字符串索引访问字符的间接迭代要简单一些.
def replace(char, word1, word2):
new_word = ""
for c1, c2 in zip(word1, word2):
if c1 == char:
new_word += char
else:
new_word += c2
return new_word
请注意,如果字符串的长度不同,那么当较短的字符串完成时,zip将正常停止.
我们可以通过用条件表达式替换if … else块来使代码更紧凑.
def replace(char, word1, word2):
new_word = ""
for c1, c2 in zip(word1, word2):
new_word += char if c1 == char else c2
return new_word
更高级的版本使用列表推导来构建所需字符的列表,使用字符串.join方法将该列表转换为字符串.
def replace(char, word1, word2):
return ''.join([char if c1 == char else c2
for c1, c2 in zip(word1, word2)])
也可以使用生成器表达式来完成此操作.
''.join(char if c1 == char else c2 for c1, c2 in zip(word1, word2))
然而,由于.join的工作方式,这比使用列表理解效率低. .join必须扫描其参数两次:第一次扫描用于计算输出字符串的总大小,第二次扫描用于实际连接.但是你无法扫描生成器两次,所以如果你通过.join生成器,它必须从生成器构建一个临时列表.