Как написать формулу Латекс в легенде сюжета с использованием Matplotlib внутри файла .py?

Я пишу script в Python (.py файл), и я использую Matplotlib для построения массива. Я хочу добавить легенду с формулой к сюжету, но я не смог это сделать. Я сделал это раньше в IPython или терминале. В этом случае напишите что-то вроде этого:

legend(ur'$The_formula$')

работал отлично. Однако это не работает, когда я вызываю свой .py script из терминала/IPython.

Ответы

import matplotlib.pyplot as plt
ax = plt.gca()  # or any other way to get an axis object
ax.plot(x, y, label=r'$\sin (x)$')

ax.legend()

import pylab

# code here

pylab.plot(x,y,'f:', '$sin(x)$')