在Python中「定期」替換字符串中的字符的最佳方式是什麼？

Question

我有一個字符串，其中一個字符（'@'）需要被一個或多個字符「按順序」和「週期性」列表中的字符替換。 所以，比如我有

'ab@cde@@fghi@jk@lmno@@@p@qrs@tuvwxy@z'

，並希望

'ab1cde23fghi1jk2lmno312p3qrs1tuvwxy2z'

爲replace_chars = ['1', '2', '3']

的問題是，在這個例子中有更多@字符串 比我有替代品。

這是我的嘗試：

result = '' 
replace_chars = ['1', '2', '3'] 
string = 'ab@cde@@fghi@jk@lmno@@@p@qrs@tuvwxy@z' 

i = 0 
for char in string: 
    if char == '@': 
     result += replace_chars[i] 
     i += 1 
    else: 
     result += char 

print(result) 

但這僅僅的作品當然，如果有不超過三個@原字符串中否則我得到IndexError。

編輯：謝謝你的答案！

Answer 1

您的代碼可以通過添加行i = i%len(replace_chars)作爲您的if子句的最後一行來解決。這樣，您將按照替換字符列表的長度從i的分區中取出剩餘部分。

較短的解決方案是使用定期吐出替換字符的生成器。

>>> from itertools import cycle 
>>> s = 'ab@cde@@fghi@jk@lmno@@@p@qrs@tuvwxy@z' 
>>> replace_chars = ['1', '2', '3'] 
>>> 
>>> replacer = cycle(replace_chars) 
>>> ''.join([next(replacer) if c == '@' else c for c in s]) 
'ab1cde23fghi1jk2lmno312p3qrs1tuvwxy2z' 

對於你的字符串s每個字符c，我們從replacer產生下一個替換字符如果字符是一個'@'，否則它只是給你原來的角色。

有關爲什麼我使用列表理解而不是生成器表達式的解釋，請參閱this。

Answer 2

發電機很有趣。

def gen(): 
    replace_chars = ['1', '2', '3'] 
    while True: 
     for rc in replace_chars: 
      yield rc 

with gen() as g: 
    s = 'ab@cde@@fghi@jk@lmno@@@p@qrs@tuvwxy@z' 
    s = ''.join(next(g) if c == '@' else c for c in s) 

正如PM2Ring所建議的，這在功能上與itertools.cycle相同。不同之處在於itertools.cycle將在內存中保存列表的額外副本，這可能不是必需的。

itertools.cycle來源：

def cycle(iterable): 
    saved = [] 
    for element in iterable: 
     yield element 
     saved.append(element) 
    while saved: 
     for element in saved: 
       yield element 

Answer 3

你也可以讓你的索引邏輯，一旦你使用取模，使用列表補償使用itertools.count跟蹤你在哪裏：

from itertools import count 

cn, ln = count(), len(replace_chars) 

print("".join([replace_chars[next(cn) % ln] if c == "@" else c for c in string])) 

ab1cde23fghi1jk2lmno312p3qrs1tuvwxy2z 

Answer 4

我想最好不要逐個字符地迭代，特別是對於沒有@的冗長部分的長字符串。

from itertools import cycle, chain 

s = 'ab@cde@@fghi@jk@lmno@@@p@qrs@tuvwxy@z' 
replace_chars = ['1', '2', '3'] 
result = ''.join(chain.from_iterable(zip(s.split('@'), cycle(replace_chars))))[:-1] 

我不知道如何有效地殺死最後一個字符[:-1]。