matplotlib colormap用法

Matplotlib 是一个 Python 的 2D 绘图库,它可以轻松地在各种平台上创建高质量的图像,并且可以输出到多种格式的文件中。其中之一就是 colormap,用于给数据图像上色。

colormap 的使用方法很简单,你可以在绘图函数中使用 cmap 参数来指定 colormap。例如,使用 imshow 函数绘制热图:

import matplotlib.pyplot as plt
import numpy as np
data = np.random.rand(10,10)
plt.imshow(data, cmap='hot')
plt.colorbar()
plt.show()

Matplotlib 默认提供了很多 colormap,例如:

  • 'hot'
  • 'cool'
  • 'gray'
  • 'jet'
  • 'viridis'
  • 'plasma'
  • 'inferno'
  • 'magma'
  • 你可以在这里查看更多colormaps的预览 matplotlib.org/stable/tuto…

    除了使用预定义的 colormap,你还可以使用自定义 colormap。例如:

    import matplotlib.colors as mcolors
    cmap = mcolors.LinearSegmentedColormap.from_list(
        'my_colormap', [(0, 'red'), (1, 'blue')], N=256)
    plt.imshow(data, cmap=cmap)
    plt.colorbar()
    plt.show()
    

    这样就可以自定义colormap了.

  •