用随机项替换 Python 字符串

Question

String replacement in Python并不难，但我想做一些特别的事情:

teststr = 'test test test test'
animals = ['bird','monkey','dog','fox']
#replace 'test' with random item from animals
finalstr = ['dog fox dog monkey']

我写了一个非常低效的版本:

from random import choice
import string
import re

teststr = 'test test test test'
animals = ['bird','monkey','dog','fox']

indexes = [m.start() for m in re.finditer('test', 'test test test test')]
#indexes = [0, 5, 10, 15]
for i in indexes:
    string.replace(teststr, 'test', choice(animals), 1)

#Final result is four random animals
#maybe ['dog fox dog monkey']

它有效，但我相信有一些我不熟悉的使用 REGULAR EXPRESSION 的简单方法。

Answer 1

使用 re.sub callback :

import re
import random

animals = ['bird','monkey','dog','fox']

def callback(matchobj):
    return random.choice(animals)

teststr = 'test test test test'
ret = re.sub(r'test', callback, teststr)
print(ret)

产量(例如)

bird bird dog monkey

re.sub 的第二个参数可以是字符串或函数(即回调)。如果它是一个函数，它会在正则表达式模式的每个非重叠出现时被调用，并且它的返回值被替换为匹配的字符串。