Легенда Matplotlib с переносами строк - PullRequest
Новая
       40

Легенда Matplotlib с переносами строк

0 голосов
/ 09 апреля 2019

Я пытаюсь добавить свои коэффициенты регрессии в виде формул LaTeX в условные обозначения сюжетов, которые состоят из нескольких строк: 

fig, ((plt1, plt2, plt3), (plt4, plt5, plt6)) = plt.subplots(2, 3, figsize=(22,10), sharex='col', sharey='row')

plot1, = plt1.plot('Normalized Times','Mean', linestyle='None', marker='o', color='#6E9EAF', markersize=marksize, data=Phase1_Temp)

plot1_R, = plt1.plot(Xdata_Phase1_Temp, Y_Phase1_Temp_Pred, linewidth=width_line, color=Orange)

plt1.legend([plot1_R], ["$f(x) = {m}*x +{b}$".format(m=np.round(A[1],2), b=np.round(A[0],2)) "\n" "$R2 = {r}$".format(r=np.round(A[2],2))])

Когда я запускаю файл, я получаю неверный синтаксис при вызове второй метки дляодна ручка: 

  "\n" "$R2 = {r}$".format(r=np.round(A[2],2))])
       ^
SyntaxError: invalid syntax

Кто-нибудь знает, как это исправить?

Ответы [ 2 ]

0 голосов
/ 09 апреля 2019

Попробуйте использовать одну строку для форматирования 

import matplotlib.pyplot as plt

A = [5,4,3]

fig, ((ax1, ax2, ax3), (ax4, ax5, ax6)) = plt.subplots(2, 3, figsize=(22,10), sharex='col', sharey='row')

plot1, = ax1.plot([0,1], linestyle='None', marker='o', color='#6E9EAF', markersize=5)

plot1_R, = ax1.plot([0,1], linewidth=2, color="orange")

ax1.legend([plot1_R], 
           ["$f(x) = {m}*x +{b}$\n$R2 = {r}$".format(m=np.round(A[1],2), 
                                                     b=np.round(A[0],2), r=np.round(A[2],2))])

plt.show()

enter image description here

Также здесь могут пригодиться строки f, где округление выполняется на уровне форматирования. 

ax1.legend([plot1_R], [f"$f(x) = {A[1]:.2f}*x +{A[0]:.2f}$\n$R2 = {A[2]:.2f}$"])

enter image description here

0 голосов
/ 09 апреля 2019

В python вы можете объединять строки следующим образом: 

"hello " "world"

и производить "hello world".Но если вы сделаете это, это синтаксическая ошибка: 

"{} ".format("hello") "world"

Так что, если вы хотите объединить из вывода format(), используйте +: 

"{} ".format("hello") + "world"

Ив вашем случае (разрыв строки добавлен для удобства чтения): 

plt1.legend([plot1_R], [
    "$f(x) = {m}*x +{b}$".format(m=np.round(A[1],2), b=np.round(A[0],2))
    + "\n"
    + "$R2 = {r}$".format(r=np.round(A[2],2))
])
...