在matplot上显示sympy方程

Question

import matplotlib.pyplot as plt
import sympy as sp
import numpy as np
q = [3,5]
t,i = sp.symbols('t,i')
eq = t**2/(q[1]-(t/q[0])**2)
x = np.linspace(0,100,10000)
y=x**2
plt.plot(x,y)

现在，我想显示sympy方程"；eq"；在情节上。我见过很多使用tex命令显示相同方程的方法，但我特别想显示sympy方程。提前感谢

Answer 1

您可以像这样使用sympy的latex()函数：

import matplotlib.pyplot as plt
import sympy as sp
import numpy as np
q = [3,5]
t,i = sp.symbols('t,i')
eq = t**2/(q[1]-(t/q[0])**2)
x = np.linspace(0,100,10000)
y=x**2  
plt.plot(x,y, label=r"$" + f"{sp.latex(eq)}" + r"$")
plt.legend()
plt.show()