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()