为何np.ones生成黑色图像而非预期白色？numpy图像可视化问题

嘿，这个问题我之前也踩过坑！核心原因在于Matplotlib的imshow()函数对不同数据类型数组的默认颜色范围（vmin/vmax）处理逻辑，咱们一步步拆解：

为啥全1的int8数组显示成黑色？ #

np.int8类型的取值范围是**-128到127**，而imshow()默认会把数组数据类型的最小值映射为颜色映射的最低点（cmap='gray'里就是黑色），最大值映射为最高点（白色）。

你生成的全1数组，所有值都是1，和int8的最大值127比起来，简直可以忽略不计，所以imshow()自动把它映射成了接近黑色的颜色，自然不符合你的预期。

再对比下另外两个场景：

• 全0的int8数组：所有值一致，直接映射为黑色，符合预期；
• 随机0/1数组：我猜你用np.random.randint生成时没指定dtype=np.int8，默认是int64类型，它的取值范围远大于0-1，所以imshow()会自动把0对应黑色、1对应白色，显示正常。

怎么修复？ #

给你两个简单的解决办法：

方法1：手动指定颜色范围 #

在imshow()里加上vmin=0和vmax=1，强制把0映射为黑色，1映射为白色：

def display_img(img):
    fig = plt.figure(figsize=(12,10))
    ax = fig.add_subplot(111)
    ax.imshow(img, cmap='gray', vmin=0, vmax=1)

方法2：转换数组数据类型 #

把int8数组转成float类型，或者用uint8类型并缩放（uint8范围是0-255，全255才是纯白）：

# 转成float类型，直接用0-1映射
ones_float = np.ones((600,600), dtype=np.int8).astype(float)
display_img(ones_float)

# 或者用uint8，把1放大到255得到纯白
ones_white = np.ones((600,600), dtype=np.uint8) * 255
display_img(ones_white)

验证小技巧 #

如果你想确认是颜色范围的问题，可以跑这段代码看看imshow自动设置的vmin和vmax：

img = np.ones((600,600), dtype=np.int8)
fig, ax = plt.subplots()
im = ax.imshow(img, cmap='gray')
print(f"自动设置的vmin: {im.get_clim()[0]}, vmax: {im.get_clim()[1]}")

输出会是自动设置的vmin: -128, vmax: 127，这下就彻底明白为啥1显示成黑色啦！

内容的提问来源于stack exchange，提问作者Enes Uğuroğlu