matplotlib显示灰度图报错TypeError: Invalid shape (4,) for image data如何解决

错误根源 #

TypeError: Invalid shape (4,) for image data 报错的核心原因是：plt.imshow() 要求输入的灰度图必须是形状为(高度, 宽度)的二维数组，而你当前计算得到的gray是长度为4的一维数组，不符合入参规则。
从你给出的gray输出结果可以反推，你用imageio.imread读取到的pic本身就不是常规RGB图像的(高度, 宽度, 3)三维结构，才会导致灰度计算后得到一维结果。

排查与修复步骤 #

• 第一步先确认图片读取是否正常，在读取代码后增加维度打印逻辑：

pic = imageio.imread("my2.jpg")
print(pic.shape) # 正常彩色jpg图片输出应为类似 (1080, 1920, 3) 的三元组

如果打印结果不是三元组，优先排查以下问题：

1. 图片路径是否正确，文件是否损坏
2. 是否有之前运行的代码覆盖了pic变量
3. imageio版本兼容性问题，高版本imageio的imread接口已迁移，推荐换用兼容性更强的PIL读取方式：

from PIL import Image
import numpy as np
# 强制转换为三通道RGB，避免RGBA四通道图片干扰
pic = np.array(Image.open("my2.jpg").convert("RGB"))

• 确认图片维度正常后，原灰度转换逻辑即可正常运行，也可以简化cmap参数的写法：

gray = np.dot(pic[... , :3] , [0.299 , 0.587, 0.114]) 
# 无需手动调用get_cmap，直接传字符串即可识别
plt.imshow(gray, cmap='gray')
plt.show() # 补充此行才会弹出显示窗口

• 更简单的替代方案：直接用PIL内置的灰度转换逻辑，无需手动计算权重：

from PIL import Image
import matplotlib.pyplot as plt

# 直接转为单通道灰度图
gray_img = Image.open("my2.jpg").convert("L")
plt.imshow(gray_img, cmap="gray")
plt.show()

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