如何在离散的颜色图中使用不同的颜色而不是RGB

Question

我想在离散的颜色图中使用不同的颜色，但是当我可以将“绿色”更改为“白色”时，我的代码不起作用。它给出以下错误：

  

KeyError：'绿色'

在下面的代码中应该使用“白色”而不是“绿色”来做什么？顺便说一句，该代码可以正常使用“绿色”颜色。

def cmap_discretize(cmap, N):
    if type(cmap) == str:
        cmap = get_cmap(cmap)
    colors_i = np.concatenate((np.linspace(0, 1., N), (0.,0.,0.,0.)))
    colors_rgba = cmap(colors_i)
    indices = np.linspace(0, 1., N+1)
    cdict = {}
    for ki,key in enumerate(('red','green','blue')):
        cdict[key] = [ (indices[i], colors_rgba[i-1,ki], colors_rgba[i,ki]) for i in range(N+1) ]

    return matplotlib.colors.LinearSegmentedColormap(cmap.name + "_%d"%N, cdict, 1024)