Matplotlib科研绘图之同一个figure中画渐变色的多条线条


最近发现网上很多的满足画多条线条,下一条线条颜色对上一条渐变的教程都过时了,现在写一个matplolib 3.5.2可用的教程。

1 颜色循环值cycler的设置

在以前的一个教程中使用set_color_cycle来控制颜色循环:

import matplotlib.pyplot as plt
import numpy as np
x = np.arange(10)
plt.gca().set_color_cycle(['red', 'green', 'blue', 'yellow'])
plt.plot(x, x)
plt.plot(x, 2 * x)
plt.plot(x, 3 * x)
plt.plot(x, 4 * x)
plt.legend(['y = x', 'y = 2x', 'y = 3x', 'y = 4x'], loc='upper left')
plt.show()

然而plt.gca()中已经没有set_color_cycle()函数了。

现在应该更改如下:

import matplotlib.pyplot as plt
import matplotlib as mpl
import numpy as np
from matplotlib import cycler

x = np.arange(10)
myCycler = cycler(color=['red', 'green', 'blue', 'yellow'])
plt.gca().set_prop_cycle(myCycler) # 修改在这里
plt.plot(x, x)
plt.plot(x, 2 * x)
plt.plot(x, 3 * x)
plt.plot(x, 4 * x)
plt.legend(['y = x', 'y = 2x', 'y = 3x', 'y = 4x'], loc='upper left')
plt.show()

2 设置颜色循环为渐近色

现在知道可以通过cycler来控制循环问题就简单多了,我们通过colormap获取一个蓝色的渐近色,然后设置到cycler,赋值给mpl.rcParams[‘axes.prop_cycle’]进行画图。

import matplotlib.pyplot as plt
import matplotlib as mpl
import numpy as np
from matplotlib import cycler
from matplotlib import cm
colormap = cm.get_cmap(name='Blues') # 获取蓝色渐近色
cycles = 4
c = [colormap(i) for i in np.linspace(0, 1, cycles)]
# mpl.rcParams['axes.prop_cycle'] = cycler(color=c) # 这个设置方法会影响之后的作图
myCycler = cycler(color=c)
plt.gca().set_prop_cycle(myCycler)
x = np.arange(10)
for i in range(cycles):
    plt.plot(x, i*x)
norm = mpl.colors.Normalize(vmin=1, vmax=cycles)
cb = plt.colorbar(mpl.cm.ScalarMappable(norm=norm, cmap=colormap),ax=plt.gca()) # 设置colormap
cb.set_ticks([i for i in range(1, cycles+1)])
plt.show()

效果如下:
在这里插入图片描述

  • 4
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值