python - 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')

图 1:

图2:

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

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')

图4:

图5: