Matplotlib:将颜色条添加到不可映射的对象

时间:2017-05-05 13:08:43

标签: python matplotlib

我有一系列代表变量变化的行;每个都有独特的颜色。出于这个原因,我想在绘图旁边添加一个颜色条。所需的输出如下所示。

问题是plot是不可映射的对象,即必须手动添加颜色条。我认为我目前的解决方案(下面)是次优的,因为它涉及我不感兴趣控制的尺寸参数。我更喜欢与可映射对象类似的解决方案(当前解决方案下面的示例)。

期望的输出

Desired output

当前解决方案

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

1 个答案:

答案 0 :(得分:9)

您可以定义自己的ScalarMappable并使用它,就像它在图中一样 (请注意,我将numbero f颜色更改为21以使其具有良好的0.1)间距

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

x = np.linspace(0, 5, 100)
N = 21
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')

norm = mpl.colors.Normalize(vmin=0,vmax=2)
sm = plt.cm.ScalarMappable(cmap=cmap, norm=norm)
sm.set_array([])
plt.colorbar(sm, ticks=np.linspace(0,2,N), 
             boundaries=np.arange(-0.05,2.1,.1))


plt.show()

enter image description here