本代码文件是使用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()