Jaki jest najlepszy sposób "okresowego" zastępowania znaków w łańcuchu w Pythonie?

Question

Mam ciąg znaków, w którym znak ("@") musi być zastąpiony przez znaki z listy jednego lub więcej znaków "w kolejności" i "okresowo". Tak na przykład mam

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

i chcą

'ab1cde23fghi1jk2lmno312p3qrs1tuvwxy2z'

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

Problemem jest to, że w tym przykładzie jest więcej @ w ciągu niż mam zamienniki.

To moja próba:

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) 

ale działa to tylko oczywiście jeśli istnieje nie więcej niż trzy @ w oryginalnym ciąg a poza tym mam IndexError.

Edytuj: Dziękujemy za odpowiedzi!

Answer 1

Twój kod można naprawić, dodając wiersz i = i%len(replace_chars) jako ostatni wiersz klauzuli if. W ten sposób otrzymasz resztę z podziału i według długości listy zastępczych znaków.

Krótsze rozwiązanie polega na użyciu generatora, który okresowo wypluwa zastępcze znaki.

>>> 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' 

Dla każdego znaku c w Twoim ciąg s, otrzymujemy kolejną zastępczą znak z generatora replacer jeśli postać jest '@', w przeciwnym razie po prostu daje oryginalny charakter.

Aby uzyskać wyjaśnienie, dlaczego użyłem rozumienia listy zamiast wyrażenia generatora, przeczytaj this.

Answer 2

Generatory są zabawne.

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) 

Zgodnie z sugestią PM 2Ring, funkcjonalnie jest taka sama jak itertools.cycle. Różnica polega na tym, że itertools.cycle będzie przechowywać dodatkową kopię listy w pamięci, która może nie być konieczna.

itertools.cycle źródło:

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

Answer 3

Można również zachować swoją logikę indeksu raz użyć modulo podstawie listy komp za pomocą itertools.count śledzić, gdzie jesteś:

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

myślę lepiej nie powtarzać znaków po znaku, szczególnie w przypadku długich ciągów z długimi częściami bez @.

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] 

Nie wiem, jak skutecznie zabić ostatni znak [:-1].