Python列表推导式内直接使用plt.imread报错原因排查

这个异常和列表推导式的执行逻辑没有任何关联，本质是你的目录遍历流程没有过滤非图像文件，导致非图像格式的文件被传给了图像读取接口。

两种写法的行为差异 #

• 你之前能正常跑通的流程：先拼接所有文件路径存成列表，后续手动逐个取路径调用plt.imread、展示图像时，大概率是在操作过程中无意识跳过了非图像文件，甚至可能只挑选了合法的图像文件做读取测试，从未把那个.mat格式的MATLAB标注文件传给读取函数，因此全程没有触发解析异常。
• 你把plt.imread直接集成到列表推导式后：推导式会无差别对遍历到的每一个目录项拼接的完整路径执行读取操作，不管这个路径对应的是jpg图像、png截图，还是混在目录里的.mat标注文件、说明txt文档，只要路径存在就直接丢给plt.imread。碰到根本不属于图像格式的.mat文件时，PIL后端无法识别文件编码格式，就会抛出UnidentifiedImageError。

常见认知误区 #

列表推导式本质就是带返回值的简化for循环，不存在特殊的执行逻辑偏差：

• 你把推导式里的读取操作拆写成普通for循环，只要不对遍历的文件做类型过滤，在碰到那个.mat文件时，会抛出完全一模一样的异常。
• 你之前觉得两种写法有逻辑差异，本质是因为之前的手动操作流程没有触碰到非图像文件，而非推导式本身有执行bug。

可直接复用的修正方案 #

• 在列表推导式内部增加文件后缀过滤逻辑，只处理合法格式的图像文件，示例代码：

  import os
  import matplotlib.pyplot as plt
  
  image_dir = "/content/sample_data/images"
  # 按需补充你目录里存在的图像格式后缀
  valid_image_exts = (".jpg", ".jpeg", ".png")
  
  images = [
      plt.imread(os.path.join(image_dir, filename))
      for filename in os.listdir(image_dir)
      # 过滤掉所有非合法图像格式的文件
      if filename.lower().endswith(valid_image_exts)
  ]
  

• 如果需要保留文件路径列表做后续处理，可以分两步做过滤，避免重复判断：

  # 第一步先筛出所有合法图像的完整路径
  valid_image_paths = [
      os.path.join(image_dir, filename)
      for filename in os.listdir(image_dir)
      if filename.lower().endswith(valid_image_exts)
  ]
  # 第二步批量读入图像到数组列表
  images = [plt.imread(path) for path in valid_image_paths]
  

注意：plt.imread本身依赖PIL做图像解析，不具备识别、跳过非图像文件的能力。无论你在什么语法结构下调用这个函数，只要传入的文件不是受支持的图像格式，都会触发相同的解析异常。

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