为什么 imshow 不能正确显示一个充满矩阵的矩阵？

Question

这是一个简单的可重现代码：

import numpy as np
import matplotlib.pyplot as plt

plt.figure()
img = np.ones((3,3))
plt.imshow(img)
plt.figure()
plt.imshow(np.array([1,1,1,1,0,1,1,1,1]).reshape(3,3))

它会给你如下图：

即使你使用

plt.imshow(img.astype(float)) or
plt.imshow(img*255)

它仍然是相同的图像。

我只需要 Matplotlib 将第一张图像作为黄色图像给我，就像第二张图像一样。

Answer 1

显式使用imshow的vmin和vmax参数：

当使用标量数据且没有明确的 norm 时，vmin 和 vmax 定义颜色图覆盖的数据范围。默认情况下，颜色图覆盖所提供数据的完整值范围。如果使用了 norm 参数，vmin、vmax 将被忽略。

import numpy as np
import matplotlib.pyplot as plt

plt.figure()
img = np.ones((3, 3))
plt.imshow(img, vmin=0, vmax=1)
plt.figure()
plt.imshow(np.array([1, 1, 1, 1, 0, 1, 1, 1, 1]).reshape(3, 3))
plt.show()

对于第二个示例，数据范围是[0 ... 1]，因此默认情况下颜色会相对于该范围进行缩放。但是，对于第一个示例，[0 ... 1] 的所需数据范围无法从数据本身（全部）中提取，因此您必须明确提供该信息。

希望有帮助！

----------------------------------------
System information
----------------------------------------
Platform:    Windows-10-10.0.16299-SP0
Python:      3.8.1
Matplotlib:  3.2.0rc3
NumPy:       1.18.1
----------------------------------------