日本免费高清视频-国产福利视频导航-黄色在线播放国产-天天操天天操天天操天天操|www.shdianci.com

學無先后,達者為師

網站首頁 編程語言 正文

Python寫一個字符串數字后綴部分的遞增函數_python

作者:Moelimoe ? 更新時間: 2022-06-01 編程語言

需求:

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

對于更多的正則表達式使用方法,可以參考之前的這篇文章

原文鏈接:https://blog.csdn.net/Moelimoe/article/details/122747365

欄目分類
最近更新