Python写一个字符串数字后缀部分的递增函数_python

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

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