如何從 function 返回 pyplot 軸，以便可以將其設置為不同 function 中的軸？

Question

我有一個繪制基本直方圖的 function。 I want this function to return the figure/axis of this histogram such that I can, in a different function, call this histogram function and set the return value as an axis in the new function. 使這一點更清楚的示例：

def histogrammer():
    fig, ax = plt.subplots()
    ax.hist(data)
    return ax

def function2():
    fig, axs = plt.subplots(2)
    axs[0].plot(x, y)
    axs[1] = histogrammer()
    axs[1].axvline(X_0)

這樣做實際上並沒有將 axs[1] 設置為直方圖儀 function 的軸，因為它沒有在 X_0 處添加垂直線。 然而，它確實會導致 histogrammer() 被打印到屏幕上。 如果重要的話，這是 Python 2.7