如何使用Python更改抓取图像的名称？

import os

image_dir = '指定目录路径'
image_files = [f for f in os.listdir(image_dir) if f.endswith('.jpg') or f.endswith('.png')]

for i, file in enumerate(image_files):
    old_name = os.path.join(image_dir, file)
    new_name = os.path.join(image_dir, f'new_image_{i}.jpg')  # 新的文件名格式
    os.rename(old_name, new_name)

import os

def change_image_names(image_dir):
    image_files = [f for f in os.listdir(image_dir) if f.endswith('.jpg') or f.endswith('.png')]
    for i, file in enumerate(image_files):
        old_name = os.path.join(image_dir, file)
        new_name = os.path.join(image_dir, f'new_image_{i}.jpg')  # 新的文件名格式
        os.rename(old_name, new_name)

# 调用函数并传入图像目录路径
change_image_names('指定目录路径')