在matplotlib中註釋數據範圍

Question

如何註釋我的一系列數據？例如，假設從x = 5到x = 10的數據大於某個截止點，我怎麼能在圖上表明這一點。如果我是手工註釋，我會在範圍上方畫一個大支架，並在支架上面寫上我的註釋。

我看到的最接近的方法是使用arrowstyle='<->'和connectionstyle='bar'，使兩個箭頭指向數據邊緣並用連線連接它們的尾部。但這並不是正確的做法;您爲註釋輸入的文本將在箭頭之下結束，而不是在條形上方。

這裏是我的嘗試，它的結果一起：

annotate(' ', xy=(1,.5), xycoords='data', 
      xytext=(190, .5), textcoords='data', 
      arrowprops=dict(arrowstyle="<->", 
          connectionstyle="bar", 
          ec="k", 
          shrinkA=5, shrinkB=5, 
          ) 
      ) 

我試圖解決的另一個問題是，標註支架的田字並沒有真正說清楚，我突出顯示範圍（不同於例如大括號）。但是，我認爲在這一點上這只是個挑剔。

Answer 1

你可以只是包裝了這一切的功能：

def add_range_annotation(ax, start, end, txt_str, y_height=.5, txt_kwargs=None, arrow_kwargs=None): 
    """ 
    Adds horizontal arrow annotation with text in the middle 

    Parameters 
    ---------- 
    ax : matplotlib.Axes 
     The axes to draw to 

    start : float 
     start of line 

    end : float 
     end of line 

    txt_str : string 
     The text to add 

    y_height : float 
     The height of the line 

    txt_kwargs : dict or None 
     Extra kwargs to pass to the text 

    arrow_kwargs : dict or None 
     Extra kwargs to pass to the annotate 

    Returns 
    ------- 
    tuple 
     (annotation, text) 
    """ 

    if txt_kwargs is None: 
     txt_kwargs = {} 
    if arrow_kwargs is None: 
     # default to your arrowprops 
     arrow_kwargs = {'arrowprops':dict(arrowstyle="<->", 
          connectionstyle="bar", 
          ec="k", 
          shrinkA=5, shrinkB=5, 
          )} 

    trans = ax.get_xaxis_transform() 

    ann = ax.annotate('', xy=(start, y_height), 
         xytext=(end, y_height), 
         transform=trans, 
         **arrow_kwargs) 
    txt = ax.text((start + end)/2, 
        y_height + .05, 
        txt_str, 
        **txt_kwargs) 


    if plt.isinteractive(): 
     plt.draw() 
    return ann, txt 

或者，

start, end = .6, .8 
ax.axvspan(start, end, alpha=.2, color='r') 
trans = ax.get_xaxis_transform() 
ax.text((start + end)/2, .5, 'test', transform=trans)