python - 如何在 matplotlib 中获取图例位置

Question

我正在尝试获取 matplotlib 中的图例位置。似乎 Legend.get_window_extent() 应该提供这个，但无论图例位于何处，它都会返回相同的值。这是一个例子:

from matplotlib import pyplot as plt

def get_legend_pos(loc):

    plt.figure()
    plt.plot([0,1],label='Plot')
    legend=plt.legend(loc=loc)

    plt.draw()

    return legend.get_window_extent()

if __name__=='__main__':

    # Returns a bbox that goes from (0,0) to (1,1)
    print get_legend_pos('upper left')

    # Returns the same bbox, even though legend is in a different location!
    print get_legend_pos('upper right')

获取图例位置的正确方法是什么？

Answer 1

您需要将 plt.draw() 替换为

plt.gcf().canvas.draw()

或者，如果您有图形句柄，fig.canvas.draw()。这是必需的，因为图例位置仅在绘制 Canvas 时确定，在此之前它只是位于同一位置。

使用 plt.draw() 是不够的，因为绘制图例需要来自正在使用的后端的有效渲染器。