Python matplotlib：如何添加像素值范围的图例？

我正在尝试使用以下代码可视化图像。如何将图例添加到此图像以表示像素的值范围

first_layer_activation = activations[31]
print(first_layer_activation.shape)

结果：

（1,16,16,1）

---

你是指类似于

plt.colorbar（）

的东西吗？

是的，我如何添加它？通过调用

plt.colorbar（）

。它对你不起作用吗？谢谢，它很成功。

import matplotlib.pyplot as plt
plt.matshow(first_layer_activation[0, :, :, 0], aspect='auto', cmap='viridis')