The plots/layout are messy/confused/disorganized.

[nickbsb]

When plotting 6 subplots using GridSpec, I want the colorbar of the 3rd subplot on the left to be aligned (horizontally level) with the 3rd subplot on the right. However, The plots/layout are messy.
import matplotlib as mpl

gs = uplt.GridSpec(nrows=4, ncols=2, pad=1,wratios=(1, 1),
hratios=(1, 1.4,1.6,0.01))
fig = uplt.figure(span=False, refwidth=2)
ax1 = fig.add_subplot(gs[0, 0])
ax1.heatmap(
df_pc1_selfeat_r ,
cmap=cmap,
labels=True,
precision=2,norm=norm,
labels_kw={"weight": "light"},
)
ax2 = fig.add_subplot(gs[1, 0])
m2 = ax2.heatmap(
df_pc2_selfeat_r ,
cmap=cmap,
labels=True,
precision=2,norm=norm,
labels_kw={"weight": "light"},
)
ax3 = fig.add_subplot(gs[2, 0])
m3 = ax3.heatmap(
df_pc3_selfeat_r ,
cmap=cmap,
labels=True,
precision=2,norm=norm,
labels_kw={"weight": "light"},
#colorbar="b",
)
ax4 = fig.add_subplot(gs[3, 0])
plt.colorbar(mpl.cm.ScalarMappable(norm=norm, cmap=cmap),
cax=ax4,orientation="horizontal")

ax5 = fig.subplot(gs[0, 1])
ax5.plot(range(5))
ax6 = fig.subplot(gs[1, 1])
ax6.plot(range(5))
ax7 = fig.subplot(gs[2:4, 1])
ax7.plot(range(5))

If I omit the line that adds the colorbar, then there is no problem.

[nickbsb]

how to fix it?

[cvanelteren]

Hi there,

I find your question a bit hard to follow. First your example contains code I cannot run as it is not self-contained. Furthermore, your question states you have 6 plots, but your gridspec contains 8. Could you please provide a minimal and verifiable example?

[nickbsb]

import matplotlib as mpl
import ultraplot as uplt
import numpy as np
import matplotlib.pyplot as plt
gs = uplt.GridSpec(nrows=4, ncols=2, pad=1,wratios=(1, 1),
                       hratios=(1, 1.4,1.6,0.01))
data1 = np.random.rand(5, 5) 
fig = uplt.figure(span=False, refwidth=2)
ax1 = fig.add_subplot(gs[0, 0])
ax1.heatmap(
    data1 ,
    precision=2,
    labels_kw={"weight": "light"},
)
data2 = np.random.rand(7, 5) 
ax2 = fig.add_subplot(gs[1, 0])
m2 = ax2.heatmap(
    data2 ,
    precision=2,
    labels_kw={"weight": "light"},
)
data3 = np.random.rand(8, 5) 
ax3 = fig.add_subplot(gs[2, 0])
m3 = ax3.heatmap(
    data3 ,
    precision=2,
    labels_kw={"weight": "light"},
)
# ax4 = fig.add_subplot(gs[3, 0])
# plt.colorbar(m3,ax=ax3,
#         cax=ax4,orientation="horizontal",fraction=1,shrink=1)

ax5 = fig.subplot(gs[0, 1])
ax5.plot(range(5))
ax6 = fig.subplot(gs[1, 1])
ax6.plot(range(5))
ax7 = fig.subplot(gs[2:4, 1])
ax7.plot(range(5))

The plot is correct after commenting out the code

[cvanelteren]

Still don't follow. I formatted the code above. The snippet you provided heavily modifies the suplotgrid. A simple example would look like this perhaps

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

data1 = np.random.rand(5, 5)
x = y = np.arange(5)
cmap = uplt.Colormap("viko")

fig, axs = uplt.subplots(
    nrows=4,
    ncols=2,
    share=0,
)
axs[:, 0].heatmap(data1, cmap=cmap)
axs[:, 1].plot(x, y)
axs[:, 0].colorbar(cmap, loc="b")

which produces a 4x2 grid with on the left heatmaps and on the right a line

[nickbsb]

Thanks, the code works, but the colorbar isn’t aligned with the plot on the right; it sticks out a bit.

[nickbsb]

Running my code does align the colorbar correctly, but the top plot looks messed up

[cvanelteren]

Thanks, the code works, but the colorbar isn’t aligned with the plot on the right; it sticks out a bit.

You mean you want it to stick out so it covers the entire 2 columns?
Have a look at fig.colorbar

fig.colorbar(mappable, loc = "b")

Since there is no real issue with the code, I am closing the issue. Feel free to open it when the issue is regarding the codebase.

[cvanelteren]

Thanks, the code works, but the colorbar isn’t aligned with the plot on the right; it sticks out a bit.

That is normal behavior; if you want the colorbar to be inset you can use an inset colorbar. Or split up the axis manually an put the colorbar within a nested subplotgrid. Either way, this is outside the scope of ultraplot in the sense of that there is no error generated on our end. This is a end-user config issue that I cannot resolve for you.

[nickbsb]

I want to use GridSpec to create an axes specifically for the colorbar. The idea is that the combined height of this colorbar and the axes above it should match the height of the right-side axes, so that the colorbar doesn’t stick out. However, when I do this, the plot above the colorbar becomes messed up.
ok,I'll try other methods.