我有一系列代表变量变化的线;每个都有独特的颜色。因此,我想在绘图旁边添加一个颜色条。所需的输出如下所示。
问题是plot
是不可映射的对象,即必须手动添加颜色条。我认为我当前的解决方案(如下)不是最佳的,因为它涉及我没有兴趣控制的尺寸参数。我更喜欢与可映射对象类似的解决方案(当前解决方案下面的示例)。
所需输出
目前的解决方案
import numpy as np
import matplotlib as mpl
import matplotlib.pyplot as plt
x = np.linspace(0, 5, 100)
N = 20
cmap = plt.get_cmap('jet',N)
fig = plt.figure(figsize=(8,6))
ax1 = fig.add_axes([0.10,0.10,0.70,0.85])
for i,n in enumerate(np.linspace(0,2,N)):
y = np.sin(x)*x**n
ax1.plot(x,y,c=cmap(i))
plt.xlabel('x')
plt.ylabel('y')
ax2 = fig.add_axes([0.85,0.10,0.05,0.85])
norm = mpl.colors.Normalize(vmin=0,vmax=2)
cb1 = mpl.colorbar.ColorbarBase(ax2,cmap=cmap,norm=norm,orientation='vertical')
plt.show()
所需的解决方案
(显然替换imshow
)
fig,ax = plt.subplots()
cax = ax.imshow(..)
cbar = fig.colorbar(cax,aspect=10)
plt.show()