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

在这里插入图片描述

发布了36 篇原创文章 · 获赞 0 · 访问量 2万+

猜你喜欢

转载自blog.csdn.net/weixin_38102912/article/details/101293798