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了.