У меня есть проект, в котором я пытаюсь построить двумерные массивы целочисленных типов.Значения могут варьироваться от 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)
Изображение отображается правильно, но цветовая полоса неправильно отформатирована:
Почему это происходит?