Если я правильно понимаю ваш вопрос, вам нужна тепловая карта, подобная этой в примерах Bokeh:
https://docs.bokeh.org/en/latest/docs/gallery/categorical.html?highlight=heatmap
Я немного поиграл с вашим кодом и пришелс чем-то похожим. Код не выполнялся точно так, как опубликовано, поэтому я не уверен, что вы видели, но вам потребуются отдельные списки для меток диапазонов (так как в списках data_dict есть повторение).
from bokeh.plotting import figure, show
from bokeh.palettes import Reds
import numpy as np
data_dict = {
'state': ['New York', 'New York', 'New Mexico', 'New Mexico'],
'feature': ['poverty', 'unemployment', 'poverty', 'unemployment'],
'colors': [Reds[9][0], Reds[9][1],
Reds[9][2], Reds[9][3]]
}
x_labels = ['poverty', 'unemployment']
y_labels = ['New York', 'New Mexico']
p = figure(title="testing",tools="hover", toolbar_location=None, x_range=x_labels, y_range=y_labels)
p.rect('feature', 'state', source=data_dict,
color='colors', width=1, height=1)
p.grid.grid_line_color = None
p.axis.axis_line_color = None
p.axis.major_tick_line_color = None
p.axis.major_label_text_font_size = "10pt"
p.axis.major_label_standoff = 0
p.xaxis.major_label_orientation = np.pi/3
show(p)