如何在图形标签中包含方形符号

Question

我想在我的图表中画出R2的值，但我不知道如何将“2”写成指数。我该怎么做？下面您可以找到代码和绘图

plt.figure()
plt.plot((y_train1),(y_train_pred),'.', color='darkviolet', alpha=1, marker='o', markersize = 2, markeredgecolor = 'black', markeredgewidth = 0.1)
plt.plot((np.array((-0.1,7))),(np.array((-0.1,7))),'-', color='magenta')
plt.xlabel('True')
plt.ylabel('Predicted')
plt.title('Train')
plt.text(5, 1, "R2 score EO: {:0.2f}".format(r2_train_EO), style='italic')

Answer 1

您应该将pyplot配置为使用Latex：

from matplotlib import rc
rc('text', usetex=True) 

然后在标签上写上R^2

编辑：如果您还想控制字体，建议您执行上述操作。如果不需要，只需在文本中写“$R^2$”——这也是Latex

Answer 2

虽然参考latex的答案是可行的，但要为这么小的改动安装latex是一件很麻烦的事。这可以通过在标签字符串中使用unicode方形符号来实现：

plt.text(5, 1, u"R\u00b2 score EO: {:0.2f}".format(r2_train_EO), style='italic')