Python Matplotlib，添加多行文字。添加可以跟随曲线的文本

我将文本添加到绘图中，在每一行中进行编码，然后调整它，使其看起来像样，增加或减少宽度，或更改位置。但是，有没有一种方法可以让Python知道文本的位置和设置方式？然后我可以添加文本，Python将计算出细节

例如，请看下图：

在图中，我在左上角有3行文本，在绘图行上方有一行

我不得不调整3条线以获得合适的间距。这不是一项困难的任务，但如果我可以说这里是文本，这里是位置，然后Python以适当的间距堆叠它，这将很容易

对于单线，我必须进行调整，使其不在线上，并降低线。对于本例，是否可以告诉python我希望文本位于绘图上方，80%位于行下方

我习惯于

LaTeX

，在这里我可以在不硬编码坐标的情况下进行调整。优点是

(1) if I want to change the location, I can change the percentage shift and not the coordinate.

(2) if the line is angled, the text will adjust to the line.

（2）的优点是，我试图将文本放在图形的顶部，该部分随直线向上倾斜

这是可以做到的，还是我要求的太多？如果是，我该怎么做

以下是实现该图的代码：

import numpy as np
import pylab

r1 = 1  #  AU Earth                                                                 
r2 = 1.524  #  AU Mars                                                              
deltanu = 75 * np.pi / 180  #  angle in radians                                     
mu = 38.86984154054163

c = np.sqrt(r1 ** 2 + r2 ** 2 - 2 * r1 * r2 * np.cos(deltanu))   
s = (r1 + r2 + c) / 2
am = s / 2


def g(a):
    alphag = 2* np.pi - 2 * np.arcsin(np.sqrt(s / (2 * a)))
    return (np.sqrt(a ** 3 / mu)
            * (alphag - betag - (np.sin(alphag) - np.sin(betag))))


def f(a):
    alpha = 2 * np.arcsin(np.sqrt(s / (2 * a)))
    beta = 2 * np.arcsin(np.sqrt((s - c) / (2 * a)))
    return (np.sqrt(a **3 / mu) * (alpha - betag - (np.sin(alpha)
                                                      - np.sin(betag))))


betag = -2 * np.arcsin(np.sqrt((s - c) / (2 * a)))
a = np.linspace(am, 2, 500000)

a = np.linspace(am, 2, 500000)

fig = pylab.figure()
ax = fig.add_subplot(111)
ax.plot(a, f(a), color = '#000000')
ax.plot(a, g(a), color = '#000000')
pylab.xlim((0.9, 2))
pylab.ylim((0, 2))

pylab.xlabel('Semi-major Axis $a$ in AU')
pylab.ylabel('Time of Flight in Years')
pylab.text(1, 1.8, '$r_1 = 1.0$ AU', fontsize = 11, color = 'r')
pylab.text(1, 1.7, '$r_2 = 1.524$ AU', fontsize = 11, color = 'r')
pylab.text(1, 1.6, '$\\Delta \\nu = 75^{\\circ}$', fontsize = 11,
           color = 'r')
pylab.text(1.75, 0.35, '$\\alpha = \\alpha_0$', fontsize = 11,
           color = 'r')
pylab.savefig('lamberttransferties.eps', format = 'eps')
pylab.show()

---

您可以使用行分隔符

\n

：

 pylab.text(1, 1.5, '$r_1 = 1.0$ AU\n' +\
                    '$r_2 = 1.524$ AU\n' +\
                    '$\\Delta \\nu = 75^{\\circ}$', fontsize = 11, color = 'r')

pylab.text（）

默认情况下使用数据坐标，但您可以在左下角使用相对位置

（0,0）

，在右上角使用相对位置

（1,1）

，传递参数

变换。请参见此示例：

pylab.text(0.6, 0.75, 'using axis coords', transform=ax.transAxes)

参数：
垂直对齐
和
水平对齐
也可以极大地帮助您。假设您想在每个角落放置一个文本：

pylab.text(1.,1.,'top-right', transform=ax.transAxes,
           horizontalalignment='right', verticalalignment='top')

pylab.text(0.,0.,'bottom-left', transform=ax.transAxes,
           horizontalalignment='left', verticalalignment='bottom')


要根据数据自动计算与文本的角度，可以采用以下方法：


检测数据最近点
找到最近点附近的a序列并使用该序列拟合曲线（以下示例使用四阶曲线）
计算要放置文本的点处的导数
例如，如果使用
ax.ax（'scaled'）
，则不需要使用
ax.axis（'scaled'）
）纠正方向性

该算法可实现如下：

def rtext(line,x,y,s, **kwargs):
    from scipy.optimize import curve_fit
    xdata,ydata = line.get_data()
    dist = np.sqrt((x-xdata)**2 + (y-ydata)**2)
    dmin = dist.min()
    TOL_to_avoid_rotation = 0.3
    if dmin > TOL_to_avoid_rotation:
        r = 0.
    else:
        index = dist.argmin()
        xs = xdata[ [index-2,index-1,index,index+1,index+2] ]
        ys = ydata[ [index-2,index-1,index,index+1,index+2] ]
        def f(x,a0,a1,a2,a3):
            return a0 + a1*x + a2*x**2 + a3*x**3
        popt, pcov = curve_fit(f, xs, ys, p0=(1,1,1,1))
        a0,a1,a2,a3 = popt
        ax = pylab.gca()
        derivative = (a1 + 2*a2*x + 3*a3*x**2)
        derivative /= ax.get_data_ratio()
        r = np.arctan( derivative )
    return pylab.text(x, y, s, rotation=np.rad2deg(r), **kwargs)

以下测试示例显示了如何使用它：

ax = pylab.subplot(111)
thetas = np.linspace(0,6*np.pi,1000)
i = np.arange(len(thetas))
xdata = (1. + (3.-1.)*i/len(thetas))*np.cos(thetas)
ydata = (1. + (3.-1.)*i/len(thetas))*np.sin(thetas)
ax.plot(xdata, ydata, color = 'b')
pylab.xlabel('x')
pylab.ylabel('y')
for x, y in zip(xdata,ydata)[::25]:
    rtext(ax.lines[0], x, y,
          '$\\alpha = \\alpha_0$', fontsize = 14, color = 'r',
          horizontalalignment='center', verticalalignment='center')


更改垂直对齐的
bottom'

您可以使用轴或图形单位而不是数据单位。我知道这里有坐标，但这很好，因为它确实回答了其中一个问题。对于这段文字，我不是在寻找绘图说明功能。通过参数转换，我们可以告诉文字跟随绘图吗？@dustin我更新了答案，添加了使文字跟随绘图的函数。。。