Add colorbar that span multiple columns at a specified row in GridSpec

[Holmgren825]

Hey!

Not sure if this is my own lack of understanding or currently not possible. But I'm wondering if it is possible to add a colobar that spans multiple columns at a specific row in a grid? Using matplotlib we can do something like this

fig, axs = plt.subplots(nrows=2, ncols=2)
rng = np.random.default_rng()
data = rng.random((100, 100))
cm = axs[0, 0].pcolormesh(data)
fig.colorbar(cm, ax=axs[0, :], location="bottom")

and get

while trying something similar in ultraplot

fig, axs = uplt.subplots(nrows=2, ncols=2)
rng = np.random.default_rng()
data = rng.random((100, 100))
cm = axs[0, 0].pcolormesh(data)
fig.colorbar(cm, ax=axs[0, :], location="bottom")

gives

I think this would be helpful when sharing colormaps among columns, but not rows.

[cvanelteren]

You are correct. I don't think we support that at this moment for inside axes colorbars. On the outside is possible, e.g.

import ultraplot as uplt, numpy as np, pandas as pd

fig, axs = uplt.subplots(nrows=2, ncols=2)
rng = np.random.default_rng()
data = rng.random((100, 100))
cm = axs[0, 0].pcolormesh(data)
fig.colorbar(cm, span=(1, 2), loc="top")
uplt.show(block=1)

[cvanelteren]

Note span is not necessary as it defaults to the entire range

[cvanelteren]

But I do see a bug that loc is not being listened to when using fig.colorbar

[cvanelteren]

Found a fix for #393 but need to tool a bit for the other one, keep you posted @Holmgren825

[Holmgren825]

Nice, quick work!