如何将自定义Bitmap类的RGB图像通过Matplotlib正常显示？

问题原因 #

这个TypeError是因为你的自定义Bitmap类存储的RGB数据是object类型数组（每个元素是RGBColor对象），但matplotlib这类绘图库只支持数值类型（比如uint8）的多维数组——需要是(高度, 宽度, 3)格式的RGB数值数组才行。

解决步骤 #

1. 把Bitmap里的RGB对象转成数值数组 #

假设你的Bitmap类用二维结构（比如pixels属性）存储像素，每个元素是RGBColor对象（带r/g/b属性，取值范围0-255），可以这么提取转换：

import numpy as np
import matplotlib.pyplot as plt

# 假设你的Bitmap实例叫bitmap_obj
rgb_list = []
for row in bitmap_obj.pixels:
    pixel_row = []
    for pixel in row:
        pixel_row.append([pixel.r, pixel.g, pixel.b])
    rgb_list.append(pixel_row)

# 转成uint8类型的numpy数组——这一步是解决错误的核心
rgb_array = np.array(rgb_list, dtype=np.uint8)

2. 验证数组格式 #

转换完成后检查数组的形状和类型，确保符合绘图库要求：

print(rgb_array.shape)  # 应输出 (height, width, 3)
print(rgb_array.dtype)  # 必须是 uint8

3. 显示图像 #

用matplotlib直接显示转换后的数组：

plt.imshow(rgb_array)
plt.axis('off')  # 不需要显示坐标轴可添加此句
plt.show()

4. 后续添加Alpha通道的适配 #

之后要加入Alpha通道时，只需要在提取像素时多带上Alpha值，转换为(height, width, 4)的uint8数组即可：

rgba_list = []
for row in bitmap_obj.pixels:
    pixel_row = []
    for pixel in row:
        # 假设RGBColor扩展后带有a属性（取值0-255）
        pixel_row.append([pixel.r, pixel.g, pixel.b, pixel.a])
    rgba_list.append(pixel_row)

rgba_array = np.array(rgba_list, dtype=np.uint8)
plt.imshow(rgba_array)
plt.show()

优化建议 #

可以在Bitmap类中添加方法直接返回数值化的数组，后续调用更便捷：

class Bitmap:
    # 你的原有代码...
    
    def to_rgb_array(self):
        return np.array([[[p.r, p.g, p.b] for p in row] for row in self.pixels], dtype=np.uint8)
    
    def to_rgba_array(self):
        return np.array([[[p.r, p.g, p.b, p.a] for p in row] for row in self.pixels], dtype=np.uint8)

调用时直接使用：

plt.imshow(bitmap_obj.to_rgb_array())
plt.show()

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