在AWS Lambda中替代matplotlib imread实现图片转数组的方法问询

在AWS Lambda环境中，你可以用以下两种方式替代matplotlib.pyplot.imread()，无需额外安装第三方库（或使用AWS预装的库）：

方案一：使用Pillow（AWS Lambda预装） #

AWS Lambda的Python运行时默认包含Pillow库，这是处理图片的标准工具，能直接读取图片并转为numpy数组，和原代码的输出格式一致。

示例代码：

import boto3
from io import BytesIO
from PIL import Image
import numpy as np

s3_client = boto3.client('s3')
image_object = s3_client.get_object(Bucket=bucket_name, Key=object_key)
img_data = BytesIO(image_object['Body'].read())

# 读取图片并转为numpy数组
with Image.open(img_data) as img:
    img_array = np.array(img)

说明：

• 读取后的img_array是HWC格式的numpy数组，和matplotlib.imread()的输出结构完全兼容。
• 原代码中的img_data1是笔误，需改为img_data。

方案二：纯标准库实现（无第三方依赖） #

如果完全不想依赖任何第三方库，你可以手动解析JPEG格式，但这种方式需要处理大量图片编码细节，代码繁琐且容易出错，仅适合特殊场景。以下是极简示例（仅处理基础灰度JPEG，不推荐用于生产）：

import boto3
from io import BytesIO
import struct

s3_client = boto3.client('s3')
image_object = s3_client.get_object(Bucket=bucket_name, Key=object_key)
img_data = BytesIO(image_object['Body'].read())

# 手动解析JPEG的基础逻辑（仅示例，不完整）
def jpeg_to_array(file_obj):
    # 跳过JPEG标记段
    while True:
        marker = struct.unpack('>H', file_obj.read(2))[0]
        if marker == 0xFFDA:  # 开始扫描
            break
        length = struct.unpack('>H', file_obj.read(2))[0]
        file_obj.read(length - 2)
    # 此处省略大量解码逻辑，实际需要完整实现JPEG解码
    # 最终返回numpy数组或列表形式的像素数据
    return []

img_array = jpeg_to_array(img_data)

注意：纯标准库实现需要完整的JPEG解码逻辑，复杂度极高，生产环境优先选择方案一。

内容的提问来源于stack exchange，提问作者Thành Đỗ