как использовать разные цвета, а не RGB в отдельной цветовой карте

Question

Я хочу использовать разные цвета в отдельной цветовой карте, но мой код не работает, когда я могу изменить "зеленый" на "белый".Выдает следующую ошибку:

KeyError: 'green'

Что я должен сделать, чтобы использовать "белый", а не "зеленый" в приведенном ниже коде?Кстати, код правильно работает с «зеленым» цветом.

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)