Matplotlib：我怎么知道正在使用哪个颜色图？

Question

我玩弄了颜色图，尝试了其中的许多，并尝试在 matplotlib 和 seaborn 中制作自己的颜色图。

但是现在我想知道我使用的是哪种颜色图。我怎样才能做到这一点？有类似的命令matplotlib.whichColormap吗？

Answer 1

通常不需要找出您正在使用的颜色图，因为您自己定义了它。即调用时

plt.imshow(..., cmap="viridis")

您已经知道您正在使用“viridis”。

如果您仍然觉得从现有的 ScalarMappable 获取该信息很有用，您可以使用get_cmap()它的name属性：

import matplotlib.pyplot as plt
import numpy as np

a = np.random.rand(4,5)
fig, ax = plt.subplots()
im = ax.imshow(a, cmap="viridis")

cm = im.get_cmap()
print(cm.name)  # prints viridis