這裏是如何使之你想要做什麼修改現有的代碼。我已將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
注意,該代碼將引發IndexError: string index out of range
異常,如果word1
比word2
短。
實現此功能有多種更好的方法。首先,我們可以使用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
它必須從發生器建立一個臨時列表。
您應該修復代碼示例的縮進。 – Carpetsmoker