遍历文件夹下所有的文件找出大于500KB的图片

import os

def find_large_images(directory, size_limit_kb, output_file):
    # 打开输出文件
    with open(output_file, 'w') as f:
        # 遍历指定目录
        for root, dirs, files in os.walk(directory):
            for file in files:
                # 构建文件完整路径
                file_path = os.path.join(root, file)
                # 获取文件大小
                file_size_kb = os.path.getsize(file_path) / 1024
                # 检查文件是否为图片且大小是否超过指定限制
                if file.lower().endswith(('.png', '.jpg', '.jpeg', '.gif', '.bmp')) and file_size_kb > size_limit_kb:
                    # 将文件路径写入输出文件
                    f.write(file_path + '\n')

if __name__ == '__main__':
    # 获取用户输入的目录路径
    directory_to_search = input('请输入要遍历的目录路径: ')
    size_limit_kb = 500  # 大小限制，单位为KB
    output_file = 'large_images.txt'  # 输出文件名

    # 运行函数
    find_large_images(directory_to_search, size_limit_kb, output_file)

    print(f'查找完成，结果已保存到 {output_file}')