如何用Python的ColorThief批量获取多张图片的主色调?
批量提取图片主色调的实现方案
当然可以批量处理,以下是直接可用的代码,能遍历你桌面文件夹里的图片,提取每张的RGB主色调,并输出文件名与对应RGB值的结果,还能处理异常图片(比如损坏文件):
from colorthief import ColorThief import os # 桌面路径,跨平台兼容 desktop_path = os.path.expanduser("~/Desktop") # 支持的图片格式,可按需添加 supported_formats = ('.jpg', '.jpeg', '.png', '.bmp', '.gif') # 存储结果的列表 results = [] # 遍历桌面文件夹 for filename in os.listdir(desktop_path): # 过滤出图片文件 if filename.lower().endswith(supported_formats): file_path = os.path.join(desktop_path, filename) try: # 提取主色调 color_thief = ColorThief(file_path) dominant_color = color_thief.get_color(quality=1) results.append((filename, dominant_color)) except Exception as e: # 处理异常情况,比如损坏的图片 print(f"处理文件 {filename} 时出错: {str(e)}") # 打印结果(两列格式) print("文件名\t\tRGB主色调") print("------------------------------") for name, rgb in results: print(f"{name}\t{rgb}") # 可选:将结果保存为CSV文件,方便用表格软件查看 import csv with open(os.path.join(desktop_path, "图片主色调结果.csv"), 'w', newline='', encoding='utf-8') as f: writer = csv.writer(f) writer.writerow(["文件名", "RGB主色调"]) writer.writerows(results) print("结果已保存到桌面的「图片主色调结果.csv」文件")
说明:
- 代码会自动识别桌面下的常见格式图片,若有其他格式可自行添加到
supported_formats中 - 加入了异常处理,遇到损坏或无法读取的图片会跳过并提示错误
- 除了控制台打印两列结果,还会在桌面生成CSV文件,用Excel或WPS打开就能看到规整的表格
内容的提问来源于stack exchange,提问作者JC91
相关产品推荐
相关产品推荐

