最近发现网上很多的满足画多条线条,下一条线条颜色对上一条渐变的教程都过时了,现在写一个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()
效果如下: