如何替换matplotlib图例并保持相同的位置？

Question

我想用一个新的图例替换matplotlib图例，但保留被替换图例的相同（可能是任意）位置。

为什么下面的代码不能工作？

import matplotlib.pyplot as plt

plt.plot(range(10))
ax = plt.gca()

leg = plt.legend(['a'], loc='lower left')
bb = leg.get_bbox_to_anchor().inverse_transformed(ax.transAxes)

plt.legend(['b'], bbox_to_anchor=bb)

Answer 1

除非我有误解，否则就做：

import matplotlib.pyplot as plt

plt.plot(range(10))
ax = plt.gca()

location = 'lower left'

while 1:

    plt.legend(['a'], loc=location)
    plt.pause(1)
    plt.legend(['b'], loc=location)
    plt.pause(1)

Answer 2

要回答问题中的代码为何不起作用的问题：
图例的位置仅在绘制图形时在运行时确定。此外，未设置bbox_至_锚定，因此它将是完整的轴（（0,0,1,1））。

你可以做的是先绘制画布，然后获得窗口范围

ax.figure.canvas.draw()
bb = leg.get_window_extent().inverse_transformed(ax.transAxes)

然而，这不会给你正确的位置（只接近它）。

现在一个可能的解决方案可能不是替换图例，而是只更新它的内容：

import matplotlib.pyplot as plt

plt.plot(range(10))
ax = plt.gca()

leg = plt.legend(['a'], loc='lower left')

leg.texts[0].set_text("b")

plt.show()

否则，如果更新不是一个选项，您将需要旧图例中的bbox_to_anchor和loc参数：

import matplotlib.pyplot as plt

plt.plot(range(10))
ax = plt.gca()

leg = plt.legend(['a'], loc='lower left', bbox_to_anchor=(0.5,0.5))

bb = leg.get_bbox_to_anchor().inverse_transformed(ax.transAxes)
plt.legend(['b'], bbox_to_anchor=bb, loc=leg._loc_real) 

plt.show()

在模式设置为“展开”的情况下，此操作仍可能失败。仅当使用相同的fancybox参数创建新图例时，它才是准确的。