Matplotlib: Отображение цветовой шкалы для прерывных целых чисел

У меня есть проект, в котором я пытаюсь построить 2D-массивы целочисленных типов. Значения могут варьироваться от 0 до 4 и могут присутствовать не все.

Я начал разрабатывать свою подпрограмму построения с целочисленным массивом resr, перечисленной цветовой картой cmap и индекс цветовой карты на основе дискретных интерваловnorm:

import matplotlib.pyplot as plt
import numpy as np
import matplotlib.colors as colors
from mpl_toolkits.axes_grid1 import make_axes_locatable

resr = np.array([[0,2],[3,4]],dtype=int)

norm = colors.BoundaryNorm(np.unique(resr), len(np.unique(resr))-1)
cmap1 = ['#7fc97f', '#ffff99', '#386cb0', '#f0027f']
cmap = colors.ListedColormap(cmap1) 

fig,ax = plt.subplots()
im = ax.imshow(resr, cmap=cmap,norm=norm)
divider = make_axes_locatable(ax)
cax = divider.append_axes("right", size = "5%")
bounds = 0.5+np.arange(norm.N+1)
cb = plt.colorbar(im, cmap=cmap,norm=norm,boundaries=bounds,cax=cax)
tic_locs =1+np.arange(norm.N+1)
cb.set_ticks(tic_locs)
cb.ax.set_yticklabels(cmap1)
cb.ax.tick_params(labelsize=10)

Что возвращает ожидаемый результат:

Когда я меняю resr на тест, чтобы убедиться, что программа построения графика будет работать в других ситуациях:

resr = np.array([[0,1],[3,4]],dtype=int)

Изображение отображается правильно, но палитра неправильно отформатирована:

Почему это происходит?