5

我是一名地质学家，有一堆不同深度的钻孔。

我粗略地设置了子图的数量、宽度和高度，以根据钻孔数量和钻孔中的样本数量而变化。

在每个钻孔中都有一个我想突出显示的区域，我用 axhspan 完成了该区域。

我想做的是在钻孔（子图）之间建立关联，绘制一条连接所有钻孔的所有分区区域的顶部和底部的线。

我尝试过使用注释，但进展不大。我不太确定如何处理这个问题，并希望得到任何建议。

这是一些示例代码，以及它可能产生的图片

import numpy as np
import matplotlib.pyplot as plt
from random import randint

fig = plt.figure()

Wells=np.arange(0,10,1) #number of wells to plot

for i in Wells:
    samp=randint(50,100) #number of samples in well

    dist=0.02     #space between plots
    left=0.05     #left border
    right=0.05    #right border
    base=0.05     #bottom border
    width=((1.0-(left+right))/len(Wells))     #width of subplot
    height=(1.0-base)/(100.0/samp)            #height of subplot

    #create subplots
    ax = fig.add_axes([left+(i*width)+dist, 1.0-(base+height), width-dist, height])    #left,bottom,width,height of subplot

    #random data 
    x=np.random.random_integers(100,size=(samp))
    y=np.arange(0,len(x),1)

    #plot
    ax.plot(x,y,alpha=0.5)    

    #zone area of plot 
    zone=samp/2.5
    ax.axhspan(15, zone, color='k', alpha=0.2) #axis 'h' horizontal span

    #format
    ax.set_ylim(0,max(y))
    ax.set_xlim(0,max(x))
    ax.tick_params(axis='both',label1On=False,label2On=False)

plt.show()

在此处输入图像描述：

4

1 回答 1

4

您可以使用matplotlib.patches.ConnectionPatch来建立这种连接。

for在循环之前添加：

xys_bot = []
xys_top = []

在for循环结束时：

for i in Wells:
    #
    #...
    #
    xys_bot.append( ((x.max() - x.min())/2., 15) )
    xys_top.append( ((x.max() - x.min())/2., zone) )
    if i > 0:
        # bottom line
        p = ConnectionPatch(xyA = xys_bot[i-1], xyB = xys_bot[i],
               coordsA='data', coordsB='data',
               axesA=fig.axes[i-1], axesB=ax,
               arrowstyle='-')
        ax.add_artist(p)
        # top line
        p = ConnectionPatch(xyA = xys_top[i-1], xyB = xys_top[i],
               coordsA='data', coordsB='data',
               axesA=fig.axes[i-1], axesB=ax,
               arrowstyle='-')
        ax.add_artist(p)

plt.draw()
plt.show()
于 2013-06-29T18:18:47.997 回答