Python:位置文本框固定在角落并正确对齐

时间:2014-09-10 17:42:11

标签: python text matplotlib

我试图模仿legend中的matplotlib.pyplot方法,其中loc='lower right'可以使用text定位图例框固定并正确对齐否关于轴和盒子的内容。

使用import matplotlib.pyplot as plt # Define some names and variables to go in the text box. xn, yn, cod = 'r', 'p', 'abc' prec = 2 ccl = [546.35642, 6785.35416] ect = [12.5235, 13.643241] fig = plt.figure() ax = fig.add_subplot(111) plt.xlim(-1., 10.) plt.ylim(-1., 1.) # Generate text to write. text1 = "${}_{{t}} = {:.{p}f} \pm {:.{p}f}\; {c}$".format(xn, ccl[0], ect[0], c=cod, p=prec) text2 = "${}_{{t}} = {:.{p}f} \pm {:.{p}f}\; {c}$".format(yn, ccl[1], ect[1], c=cod, p=prec) text = text1 + '\n' + text2 ax.annotate(text, xy=(0.75, 0.9), xycoords='axes fraction', fontsize=10, bbox=dict(facecolor='white', alpha=0.8), horizontalalignment='left', verticalalignment='bottom') plt.savefig('annotate_test.png', dpi=150) 已经出局,因为这需要手动输入坐标,而我需要自动输入。

我尝试过使用annotate,它让我走了一半,但它仍然无法正常工作。

这是我到目前为止所做的:

ax.set_aspect('equal')

导致:

enter image description here

这将正确缩放以更改轴限制,但问题是:1-如果轴设置为prec=5,它将失败:

enter image description here

和2-如果文本太长会失败(这里我在上面的MWE中设置了matplotlib):

enter image description here

如何告诉loc将文本框始终放在右上角并正确对齐,这样它就不会落在图像之外(即:什么legend中有{{1}}吗?

1 个答案:

答案 0 :(得分:6)

快速而肮脏的方法是使用右对齐和顶部对齐的文本,并将其放置在距离轴角的点的固定偏移处:

import matplotlib.pyplot as plt

# Define some names and variables to go in the text box.
xn, yn, cod = 'r', 'p', 'abc'
prec = 2
ccl = [546.35642, 6785.35416]
ect = [12.5235, 13.643241]

fig = plt.figure()
ax = fig.add_subplot(111)
ax.axis([-1, 10, -1, 1])

# Generate text to write.
text1 = "${}_{{t}} = {:.{p}f} \pm {:.{p}f}\; {c}$".format(xn, ccl[0],
    ect[0], c=cod, p=prec)
text2 = "${}_{{t}} = {:.{p}f} \pm {:.{p}f}\; {c}$".format(yn, ccl[1],
    ect[1], c=cod, p=prec)
text = text1 + '\n' + text2

ax.annotate(text, xy=(1, 1), xytext=(-15, -15), fontsize=10,
    xycoords='axes fraction', textcoords='offset points',
    bbox=dict(facecolor='white', alpha=0.8),
    horizontalalignment='right', verticalalignment='top')

plt.show()

enter image description here

因为我们已经指定了顶部和右部对齐,所以它适用于您的两个边缘情况:

enter image description here

enter image description here


这样做的缺点是文本是右对齐的。理想情况下,您希望文本对齐与框对齐分开。 matplotlib.offsetbox模块有许多方法可以处理这样的事情。

如果您想模仿图例框(下至位置代码),请查看matplotlib.offsetbox.AnchoredText。 (请注意,您可以通过padborderpad kwargs调整填充等:http://matplotlib.org/api/offsetbox_api.html#matplotlib.offsetbox.AnchoredOffsetbox

import matplotlib.pyplot as plt
import matplotlib.offsetbox as offsetbox

# Define some names and variables to go in the text box.
xn, yn, cod = 'r', 'p', 'abc'
prec = 5
ccl = [546.35642, 6785.35416]
ect = [12.5235, 13.643241]

fig = plt.figure()
ax = fig.add_subplot(111)
ax.axis([-1, 10, -1, 1])

# Generate text to write.
text1 = "${}_{{t}} = {:.{p}f} \pm {:.{p}f}\; {c}$".format(xn, ccl[0],
    ect[0], c=cod, p=prec)
text2 = "${}_{{t}} = {:.{p}f} \pm {:.{p}f}\; {c}$".format(yn, ccl[1],
    ect[1], c=cod, p=prec)
text = text1 + '\n' + text2

ob = offsetbox.AnchoredText(text, loc=1)
ax.add_artist(ob)

plt.show()

enter image description here

这样做的一个缺点是调整结果的字体和框参数有点违反直觉。 AnchoredText接受字体参数字典作为prop kwarg。初始化后,可以通过patch属性调整该框。作为一个简单的例子:

ob = offsetbox.AnchoredText(text, loc=1,
                    prop=dict(color='white', size=20))
ob.patch.set(boxstyle='round', color='blue', alpha=0.5)
ax.add_artist(ob)

enter image description here