生成图片灰度值二维列表并导出至Excel的技术问询

核心功能实现思路 #

要完成需求，核心分为三步：

• 将图片转换为灰度模式，确保每个像素值为0-255的灰度值
• 按左上到右下、逐行遍历的顺序，把每行像素封装为子列表，最终生成二维列表
• 将二维列表导出至Excel文件

代码优化方向 #

针对需求，优化点集中在效率、可读性和健壮性上：

• 避免低效的逐像素循环：利用PIL内置方法批量处理像素，提升大图片处理速度
• 增加异常处理：捕获文件不存在、格式不支持等异常
• 资源安全管理：使用with语句自动释放图片资源
• 代码可读性：添加类型注解、清晰的变量命名
• 灵活配置：将输入输出路径设为可配置参数

完整优化代码 #

from PIL import Image
import pandas as pd
from typing import List

def image_to_gray_2d_list(image_path: str) -> List[List[int]]:
    """
    将图片转换为灰度值二维列表，每行作为一个子列表
    :param image_path: 输入图片路径
    :return: 灰度值二维列表，格式为[[row1_pixel1, row1_pixel2,...], [row2_pixel1,...]]
    """
    # 用with语句自动管理图片资源
    with Image.open(image_path) as img:
        # 转换为灰度模式（L模式，像素值范围0-255）
        gray_img = img.convert("L")
        width, height = gray_img.size
        
        # 批量生成二维列表：逐行读取像素，避免逐像素循环
        gray_2d_list = []
        for y in range(height):
            # 截取当前行区域并批量获取像素值
            row_pixels = list(gray_img.crop((0, y, width, y+1)).getdata())
            gray_2d_list.append(row_pixels)
        
        return gray_2d_list

def export_gray_list_to_excel(gray_list: List[List[int]], excel_path: str) -> None:
    """
    将灰度值二维列表导出至Excel
    :param gray_list: 灰度值二维列表
    :param excel_path: 输出Excel文件路径
    """
    # 利用pandas快速导出，自动处理行列对齐
    df = pd.DataFrame(gray_list)
    df.to_excel(excel_path, index=False, header=False)

if __name__ == "__main__":
    # 配置参数
    INPUT_IMAGE = "test_image.png"
    OUTPUT_EXCEL = "gray_values.xlsx"
    
    try:
        gray_data = image_to_gray_2d_list(INPUT_IMAGE)
        export_gray_list_to_excel(gray_data, OUTPUT_EXCEL)
        print(f"处理完成！灰度值已导出至 {OUTPUT_EXCEL}")
    except FileNotFoundError:
        print(f"错误：未找到图片文件 {INPUT_IMAGE}")
    except Exception as e:
        print(f"处理出错：{str(e)}")

关键细节说明 #

1. 灰度转换：使用img.convert("L")直接将图片转为灰度模式，无需手动计算RGB转灰度的公式，确保像素值在0-255区间
2. 高效行读取：通过crop截取当前行区域，再用getdata()批量获取该行像素，比逐个调用getpixel效率提升数倍
3. Excel导出：借助pandas的DataFrame和to_excel方法，无需手动处理单元格写入，代码简洁且不易出错
4. 异常处理：捕获常见错误并输出清晰提示，提升代码健壮性

内容的提问来源于stack exchange，提问作者Bart008