python - 用括号中的数字Python替换字符串中的数字

Question

我有这样一个字符串:

s = k0+k1+k1k2+k2k3+1+12

我想转换它，这样每个跟在字母(k 此处)后面的数字都被方括号包围:

k[0]+k[1]+k[1]k[2]+k[2]k[3]+1+12

这样做的好方法是什么？

我试过的:使用 replace() 函数 4 次(但它不能处理后面没有字母的数字)。

Answer 1

这是一个使用 re 模块和正则表达式 ([a-zA-Z])(\d+) 的选项，它匹配一个字母后跟数字和sub，您可以在替换中用一对括号将匹配的数字括起来:

import re
s = "k0+k1+k1k2+k2k3+1+12"

re.sub(r"([a-zA-Z])(\d+)", r"\1[\2]", s)
# 'k[0]+k[1]+k[1]k[2]+k[2]k[3]+1+12'

---

要将匹配的字母替换为大写，可以在替换位置使用 lambda 将它们转换为大写:

re.sub(r"([a-zA-Z])(\d+)", lambda p: "%s[%s]" % (p.groups(0)[0].upper(), p.groups(0)[1]), s)

# 'K[0]+K[1]+K[1]K[2]+K[2]K[3]+1+12'