plt.colorbar support for umap.plot.points

Hello, umap.plot.points does not seem to support a colour bar when colouring the points with continuous values, is this a design choice?

I’ve tried

umap.plot.points(umap_object, values = array_of_numbers)
plt.colorbar()

With no success.

I think it’d be quite handy to have such functionality

[zachschillaci27]

Hi @lucharo , not sure if your issue is still outstanding, but I recently ran into a similar problem myself. I implemented a quick workaround which I will share below.

First create a pcolormesh for the series which you will be coloring with umap:

psm = plt.pcolormesh([data_series, data_series], cmap=mpl.cm.get_cmap('viridis'))

Then create the umap points plot as usual, ensuring the data series and color theme are the same:

ax = umap.plot.points(mapper, values=data_series, theme='viridis')

Now create the colorbar from the pcolormesh and draw alongside the umap figure:

cb = plt.colorbar(psm, ax=ax)

From here you can edit the colorbar object as you please to change the label size, tick colors, etc …

Hope this helps!

[jondo]

@lc5415, would you mind showing your emulating function?