怎么用Python写一个字符串数字后缀部分的递增函数-编程学习网

本篇内容介绍了“怎么用Python写一个字符串数字后缀部分的递增函数”的有关知识，在实际案例的操作过程中，不少人都会遇到这样的困境，接下来就让小编带领大家学习一下如何处理这些情况吧！希望大家仔细阅读，能够学有所成！

需求：
Python处理重名字符串，添加或递增数字字符串后缀

对于2个重名的字符串，添加数字后缀，比如两个重复的字符串s1 = “name”, s2 = “name”，将s2的名称设置为name_1
对应3个或多个重名的字符串，数字部分实现递增，初始时s1=s2=s3=“name”，重命名后得到s1=“name”, s2=“name_1”, s3=“name_2”
要灵活处理字符串问题，使用re模块最为方便
下面是一个对带有下划线+数字的后缀字符串递增的方法

def increase_string_suffix(s, incr_num=1):
"""
带数字后缀"_d"的字符串自增方法，"name_1" 自增1 --> "name_2"
Example
-----------------
>>> s = "name_01"
>>> increase_string_suffix(s, incr_num=2)
'name_03'
"""
suffix_searched = re.search(r"(_)(\d+)$", s)
if suffix_searched:
suffix_plus_1 = re.sub(
r"(_)(\d+)$",
lambda x: f"{x.group(1)}{str(int(x.group(2)) + incr_num).zfill(len(x.group(2)))}",
s
)
else:
suffix_plus_1 = f"{s}_1"
return suffix_plus_1

例子在代码中可以直接运行注释的doc部分，参数s是要传入的字符串，incr_num是一次要增加的步数
代码中核心部分是re.sub方法，r"(_)(\d+)$"是匹配搜寻部分，()对匹配的部分进行分组，x.group(1)是()匹配到的""，x.group(2)是匹配到的数字部分，zfill往左边填0补上长度，例如01→02而不会是01→2

也可以写一个递减的，或是修改前缀的，例如

def increase_string_prefix(s, incr_num=1):
"""
带数字前缀"d-"的字符串自增方法，"1-name" 自增1 --> "2-name"
Example
-----------------
>>> s = "1-name"
>>> increase_string_prefix(s, incr_num=1)
'2-name'
"""
prefix_searched = re.search(r"^(\d+)(-)", s)
if prefix_searched:
prefix_plus_1 = re.sub(
r"^(\d+)(-)",
lambda x: f"{str(int(x.groups()[0]) + incr_num).zfill(len(x.groups()[0]))}{x.groups()[1]}",
s
)
else:
prefix_plus_1 = f"1-{s}"
return prefix_plus_1