Python - Mapas de color

import skimage.io as io

import numpy as np

import matplotlib.pyplot as plt

import cv2

a = np.ones((50,200))

b = a*100

c =a*200

d = np.block([[b],[c],[b]])

e =d.astype('uint8')

f= e

def applyCustomColorMap(b_esp):

    mapac_esp =np.zeros((256,1,3),dtype=np.uint8)

    mapac_esp[0:105,0,0]=255

    mapac_esp[105:255,0,0:2]=255

    im_color =cv2.applyColorMap(b_esp,mapac_esp)

    return im_color;

espa =applyCustomColorMap(e);

plt.imshow(espa);

def applyCustomColorMap(b_arg):

    mapac_esp =np.zeros((256,1,3),dtype=np.uint8)

    mapac_esp[0:105,0,2]=255

    mapac_esp[105:255,0,:]=255

    im_color =cv2.applyColorMap(b_arg,mapac_esp)

    return im_color;

arg =applyCustomColorMap(f);

plt.subplot

plt.imshow(arg);