import matplotlib as mpl
mpl.use('pgf')
import matplotlib.pyplot as plt
x = np.linspace(0.0, 1.0, num=100)
y = x ** 2
plt.plot(x, y, label='Data \\cite{<key>}')
plt.legend(loc=0)
plt.savefig('fig.pgf')
Latex Citation in matplotlib Legend