How to change cmap based on array value in python

Question

Current code:

        # Mask array with mask pixels set to 1 and rest set to 0
        mask = nib.load(os.path.join(mask_path, f))
        # Grey scale MRI image