python - 选择性 python 字符串替换

Question

我正在尝试替换 pi与 math.pi使用以下 Python 函数。

def cleanup(x):
  return x.replace("pi", "math.pi")

我有以下字符串:

a = "2*pi"
b = "the pink elephant"

cleanup(a) 的输出是:2*math.pi -- 这很好用!

cleanup(b) 的输出是 the math.pink elephant -- 问题:我不想改变“文本”。

有人可以帮助我吗？

Answer 1

您正在寻找正则表达式，特别是“单词边界”(\b) 断言:

import re
print re.sub(r'\bpi\b', 'math.pi', "2*pi")
print re.sub(r'\bpi\b', 'math.pi', "the pink elephant")