소개

matplotlib에서 SymPy의 수식을 LaTeX를 사용하여 아름답게 그립니다.

import numpy as np 
import matplotlib.pyplot as plt 
import sympy

# おまじない
plt.rcParams['text.usetex'] = True

준비

x  = sympy.Symbol('x')
xx = np.linspace(0,1,1000)

f  = sympy.sin(sympy.pi*x)
ff = sympy.lambdify(x,f) 

plt.figure() 
plt.plot(xx,ff(xx))

LaTeX 명령

LaTeX 명령을 사용할 때 $ 마크로 묶습니다.

plt.title('$f(x) = \sin(\pi x)$')

SymPy에서 변환하여 그리기

sympy.latex ()에서 LaTeX 명령으로 변환하여 $ 마크로 묶습니다.

plt.title('$f(x) = ' + sympy.latex(f)+'$')

코드