Я пытаюсь добавить текст в записи легенды в Python 3, используя matplotlib, чтобы я мог пометить их, например, на графике есть три разные кривые, и все они соответствуют различным значениям переменных, используемым для генерации кривых, но сейчас легенда отображает только значение переменной, а не ее имя. Например, если я строю три разные линии с разным наклоном, легенда дает значение наклона (то есть число 4), но я хочу, чтобы легенда говорила «slope = 4», а не просто 4. Мой код здесь:
def plotter(to_plot, var, *args):
"""Plots different object variables"""
list_of_objs=[arg for arg in args]
for obj in list_of_objs:
plt.plot(a.f_in, getattr(obj, to_plot))
plt.xlabel('Flux',fontsize='xx-large')
plt.ylabel(to_plot,fontsize='xx-large')
plt.xticks(fontsize='xx-large')
plt.yticks(fontsize='xx-large')
for_legend = [str(getattr(obj,var)) for obj in list_of_objs]
plt.legend(for_legend ,fontsize='xx-large')
plt.show()
Код принимает список объектов класса, которые создают разные кривые на графике. На данный момент легенда печатает только значение переменной, заданное значениями, хранящимися в for_legend, но я хочу, чтобы оно было var = for_legend. Любые советы здесь? Заранее спасибо.