python更改文件夹下图片名字

读取文件夹下的每一张图片，旧名字为line1_label.png、line2_label.png.....line100_label.png，对每一张重命名，要求新名字中含有旧名字的数字，新名字格式为line1.png....line100.png

import os
import re
import shutil


def rename_images(folder_path):
    # 检查文件夹是否存在
    if not os.path.exists(folder_path):
        print(f"文件夹 '{folder_path}' 不存在。")
        return

    # 遍历文件夹中的文件
    file_list = os.listdir(folder_path)
    image_files = [f for f in file_list if re.match(r"line\d+_label\.png", f)]

    for old_name in image_files:
        # 提取旧名字中的数字
        match = re.search(r"\d+", old_name)
        if match:
            number = match.group()
            # 构建新名字
            new_name = re.sub(r"\d+_label.png", "", old_name)
            new_name = f"{new_name}{number}.png"

            # 重命名文件
            old_path = os.path.join(folder_path, old_name)
            new_path = os.path.join(folder_path, new_name)
            shutil.move(old_path, new_path)
            print(f"重命名文件：'{old_name}' -> '{new_name}'")


# 调用函数进行重命名
folder_path = "H:\\深度学习\\seg1\\"  # 将路径替换为实际的文件夹路径
rename_images(folder_path)