带换行符的Matplotlib图例

Question

我正尝试将回归系数作为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

有人知道如何解决此问题吗？

Answer 1

在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))
])

Answer 2

考虑使用单个字符串进行格式化

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()

此外，f字符串在这里很方便，在格式化级别进行四舍五入。

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