colormap绘制渐变图

本代码文件是使用colormap绘制渐变图
开始设置好:

cNorm = colors.Normalize(vmin=0, vmax=5)
scalarMap = cmx.ScalarMappable(norm=cNorm, cmap=jet)

注意这里需要设置vmin, vmax意味着后续图片渐变的最小值和最大值
在绘图的过程中加入

colorVal = scalarMap.to_rgba(i)

即可以将相关的值 映射到相应的 rbg空间中
注意每次绘图的过程中 c只能是一个数
虽然是渐变的过程,实际上是一段一段地绘制


import numpy as np
#%matplotlib inline
import matplotlib.pyplot as plt
import mpl_toolkits.mplot3d as p3d
import matplotlib.colors as colors
import matplotlib.cm as cmx

jet = cm = plt.get_cmap('Reds') 
#这里可以选择色条方案
#e.g. Greys, 'Blues'
cNorm = colors.Normalize(vmin=0, vmax=5)
scalarMap = cmx.ScalarMappable(norm=cNorm, cmap=jet)
print(scalarMap.get_clim()) 

fig = plt.figure()
ax = fig.add_subplot(111)

#------绘图---------------------
x = np.arange(10, 15, 1)
y = np.arange(10, 15, 1)

for i in range(5):
    colorVal = scalarMap.to_rgba(i)
    print(i, colorVal)
    ax.plot(x, y + i, c = colorVal )  
plt.show() 

在这里插入图片描述

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章