matplotlib：相对于其他轴放置轴，自动更新

Question

Matplotlib有很好的文档说明如何在图形窗口中放置多组轴，但我无法弄清楚如何定义一组轴相对于另一组轴的位置的位置。例如，

import matplotlib.pyplot as plt
import numpy as np    

#Define data
x1 = np.arange(0,10,0.01)
y1 = np.sqrt(x1)
x2 = x1
y2 = 1.0/2.0 * x2**2.0

#Generate vertically stacked plots
fig = plt.figure()
ax1 = fig.add_subplot(211)
ax1.plot(x1,y1)
ax2 = fig.add_subplot(212)
ax2.plot(x2,y2)
fig.savefig('nice_stacked_plots.png')

给出以下图：

这一切都很好，但是当我改变底轴的大小时

#Change the size of the bottom plot
bbox2 = ax2.get_position()
ax2.set_position([bbox2.x0, bbox2.y0, bbox2.width, bbox2.height * 1.25])
ax2.set_ylim(0,60)
fig.savefig('overlapping_stacked_plots.png')

底部轴与顶部轴重叠

我意识到我可以随后更新顶轴的位置以消除重叠，但我想在开始时指定相对于底轴的顶轴位置，并自动更新。

例如，在annotate tutorial中，可以放置注释，然后使用OffsetFrom类将第二个注释放在第一个注释的指定偏移处。如果第一个注释移动，则第二个注释随之移动。我想用轴做类似的事情。

Answer 1

我担心我没有提供一般答案，但你知道add_axes吗？

它允许您精确定义子图的位置 - 然后很容易使一个依赖于另一个。 这是一个例子 - 正如我所说，非常适合你的任务，但也许它可能会激励你？

# General aspect of the Fig (margins)
left = 0.1
right = 0.05
width= 1.-left-right
bottom = 0.1
top = 0.05
hspace = 0.10 #space between the subplots

def placeSubplots(fig, ax2height = (1.-top-bottom-hspace)/2.):
    ax1height = 1-top-bottom-hspace-ax2height
    ax1 = fig.add_axes([left, bottom+ax2height+hspace, width, ax1height])
    ax1.plot(x1, y1)
    ax2 = fig.add_axes([left, bottom, width, ax2height])
    ax2.plot(x2, y2)
    return fig

fig1 = placeSubplots(plt.figure())
fig2 = placeSubplots(plt.figure(), ax2height=0.6)

fig1.savefig('fig1_equal_heigth.png')
fig2.savefig('fig2_ax2_taller.png')

FIG1：

fig2：

在上面，第二个轴的高度以绝对方式指定，但您也可以将子图的高度定义为它们之间的比例：

def placeSubplotsRatio(fig, ax1ax2ratio = 1.):
    subplotSpace = 1.-top-bottom-hspace
    ax1height = subplotSpace/(1.+1./ax1ax2ratio)
    ax2height = subplotSpace/(1.+ax1ax2ratio)
    ax1 = fig.add_axes([left, bottom+ax2height+hspace, width, ax1height])
    ax1.plot(x1, y1)
    ax2 = fig.add_axes([left, bottom, width, ax2height])
    ax2.plot(x2, y2)
    return fig

fig3 = placeSubplotsRatio(plt.figure()) # idem as fig1
fig4 = placeSubplotsRatio(plt.figure(), ax1ax2ratio=3.) #ax1 is 3 times taller
fig5 = placeSubplotsRatio(plt.figure(), ax1ax2ratio=0.25) #ax2 is 4 times taller

fig4.savefig('fig4_ax1ax2ratio3.png')
fig5.savefig('fig5_ax1ax2ratio025.png')

图四：

fig5：