颜色之间具有刻度标签的离散颜色条

Question

我试图用离散颜色条来绘制某些数据。我是按照给出的例子（https://gist.github.com/jakevdp/91077b0cae40f8f8244a），但问题是这个例子不工作1-1不同的间距。例如，链接中示例中的间距仅增加1，但我的数据增加了0.5。你可以看到我的代码输出。 。任何帮助，这将不胜感激。我知道我在这里失去了一些关键，但无法弄清楚。

import matplotlib.pylab as plt 
import numpy as np 

def discrete_cmap(N, base_cmap=None): 
    """Create an N-bin discrete colormap from the specified input map""" 

    # Note that if base_cmap is a string or None, you can simply do 
    # return plt.cm.get_cmap(base_cmap, N) 
    # The following works for string, None, or a colormap instance: 

    base = plt.cm.get_cmap(base_cmap) 
    color_list = base(np.linspace(0, 1, N)) 
    cmap_name = base.name + str(N) 
    return base.from_list(cmap_name, color_list, N) 
num=11 

x = np.random.randn(40) 
y = np.random.randn(40) 
c = np.random.randint(num, size=40) 


plt.figure(figsize=(10,7.5)) 

plt.scatter(x, y, c=c, s=50, cmap=discrete_cmap(num, 'jet')) 
plt.colorbar(ticks=np.arange(0,5.5,0.5)) 
plt.clim(-0.5, num - 0.5) 
plt.show() 

Answer 1

好吧，这是我发现我自己的问题的黑客攻击。我相信有一个更好的方法来做到这一点，但这对我正在做的事情有效。随意提出一个更好的方法来做到这一点。

import numpy as np 
import matplotlib.pylab as plt 

def discrete_cmap(N, base_cmap=None): 
    """Create an N-bin discrete colormap from the specified input map""" 

    # Note that if base_cmap is a string or None, you can simply do 
    # return plt.cm.get_cmap(base_cmap, N) 
    # The following works for string, None, or a colormap instance: 

    base = plt.cm.get_cmap(base_cmap) 
    color_list = base(np.linspace(0, 1, N)) 
    cmap_name = base.name + str(N) 
    return base.from_list(cmap_name, color_list, N) 

num=11 

plt.figure(figsize=(10,7.5)) 

x = np.random.randn(40) 
y = np.random.randn(40) 
c = np.random.randint(num, size=40) 


plt.scatter(x, y, c=c, s=50, cmap=discrete_cmap(num, 'jet')) 
cbar=plt.colorbar(ticks=range(num)) 
plt.clim(-0.5, num - 0.5) 
cbar.ax.set_yticklabels(np.arange(0.0,5.5,0.5)) 
plt.show() 

出于某种原因，我无法上传与上面的代码相关联的图像。上传时出现错误，因此不确定如何显示最终示例。但简单地说，我为垂直彩条的刻度标签设置了颜色条轴，并将其传递到了我想要的标签，并生成了正确的输出。