python - Matplotlib：subplot2grid 中绘图外的颜色条

Question

我试图在一些子图中绘制两个高斯分布。这是结果：

这是代码

import numpy as np
import numpy.random as rnd
import matplotlib.pyplot as plt
from mpl_toolkits.axes_grid1 import make_axes_locatable



mu_x1 = 3
sig_x1 = 2

mu_x2 = 3
sig_x2 = 0.5

N = int(1e3)

gaussian1 = rnd.normal(mu_x1, sig_x1, N)
gaussian2 = rnd.normal(mu_x2, sig_x2, N)
fig = plt.figure()

ax1 = plt.subplot2grid(shape = (3, 3),
                       loc = (1, 0),
                       rowspan = 2,
                       colspan = 2,
                       fig = fig
                       )
hist2d = ax1.hist2d(gaussian1,
                    gaussian2,
                    bins = 100,
                    density = True,
                    cmap=plt.cm.jet
                    )
fig.colorbar(hist2d[3], ax=ax1)
ax1.set_xlabel("$x_1$")
ax1.set_ylabel("$x_2$")



ax2 = plt.subplot2grid(shape = (3, 3),
                       loc = (0, 0),
                       rowspan = 1,
                       colspan = 2,
                       fig = fig
                       )
ax2.hist(gaussian1,
         bins = 100
         )
ax2.xaxis.set_ticks_position('top')
ax2.set_xlabel("$x_1$")
ax2.xaxis.set_label_position('top')
ax2.set_ylabel("Cuentas")

ax3 = plt.subplot2grid(shape = (3, 3),
                       loc = (1, 2),
                       rowspan = 2,
                       colspan = 1,
                       fig = fig
                       )
ax3.hist(gaussian2,
         bins = 100,
         orientation=u'horizontal')
ax3.yaxis.set_ticks_position('right')
ax3.yaxis.set_label_position('right')
ax3.set_xlabel("Cuentas")
ax3.set_ylabel("$x_2$")

plt.show()

它看起来不错，但不是我想要的。我希望中间子图的 x 轴与上面子图的 x 轴一样大，所以它们重合。问题是彩条。我想将颜色条移到中间子图之外来做我想做的事。我查看了一些信息，然后到达：

这里的代码

fig = plt.figure()

ax1 = plt.subplot2grid(shape = (4, 3),
                       loc = (1, 0),
                       rowspan = 2,
                       colspan = 2,
                       fig = fig
                       )
hist2d = ax1.hist2d(gaussian1,
                    gaussian2,
                    bins = 100,
                    density = True,
                    cmap=plt.cm.jet
                    )
#fig.colorbar(hist2d[3], ax=ax1)
ax4 = plt.subplot2grid(shape = (4, 3),
                       loc = (3, 0),
                       rowspan = 1,
                       colspan = 2,
                       fig = fig
                       )    
fig.colorbar(hist2d[3], cax=ax4, orientation="horizontal")
ax1.set_xlabel("$x_1$")
ax1.set_ylabel("$x_2$")
ax1.axhline(mu_x2 + sig_x2*n2,c="k")
ax1.axhline(mu_x2 - sig_x2*n2,c="k")
ax1.axvline(mu_x1 + sig_x1*n1,c="k")
ax1.axvline(mu_x1 - sig_x1*n1,c="k")



ax2 = plt.subplot2grid(shape = (4, 3),
                       loc = (0, 0),
                       rowspan = 1,
                       colspan = 2,
                       fig = fig
                       )
ax2.hist(gaussian1,
         bins = 100
         )
ax2.xaxis.set_ticks_position('top')
ax2.set_xlabel("$x_1$")
ax2.xaxis.set_label_position('top')
ax2.set_ylabel("Cuentas")
ax2.axvline(mu_x1 + sig_x1*n1,c="k")
ax2.axvline(mu_x1 - sig_x1*n1,c="k")
#ax2.set_xticks([])

ax3 = plt.subplot2grid(shape = (4, 3),
                       loc = (1, 2),
                       rowspan = 2,
                       colspan = 1,
                       fig = fig
                       )
ax3.hist(gaussian2,
         bins = 100,
         orientation=u'horizontal')
ax3.yaxis.set_ticks_position('right')
ax3.yaxis.set_label_position('right')
ax3.set_xlabel("Cuentas")
ax3.set_ylabel("$x_2$")
ax3.axhline(mu_x2 + sig_x2*n2,c="k")
ax3.axhline(mu_x2 - sig_x2*n2,c="k")

现在的问题是颜色条正在填充完整的子图，它太大并且隐藏了上面子图的刻度。

有什么办法解决吗？

标签： pythonmatplotlibsubplotcolorbar