matplotlib图例中的上标格式

时间:2018-12-14 14:42:05

标签: python matplotlib superscript

我正在使用奇妙的scipy曲线拟合进行一些曲线拟合。在绘制数据并添加图例标签以显示计算出的参数时,使用$^{}$来使位之间的上标仅在写入字符串时有效,而在从字符串格式调用时无效。即$x^{}$.format(3)格式不正确,但$x^3$格式正确。

这行得通吗?如果我向图例标签提供输入,是否还需要做其他事情?

下面的示例代码和图解。谢谢。

import matplotlib.pyplot as plt
import numpy as np
from scipy.optimize import curve_fit

x_data = np.linspace(0.05,1,101)
y_data = 1/x_data
noise = np.random.normal(0, 1, y_data.shape)
y_data2 = y_data + noise

def func_power(x, a, b):
    return a*x**b 

popt, pcov= curve_fit(func_power, x_data, y_data2)

plt.figure()
plt.scatter(x_data, y_data2, label = 'data')
plt.plot(x_data, popt[0] * x_data ** popt[1], label = ("$y = {}x^{}$").format(round(popt[0],2), round(popt[1],2)))
plt.plot(x_data, x_data**3, label = '$x^3$')
plt.legend()
plt.show()

enter image description here

1 个答案:

答案 0 :(得分:3)

为使MathText解释大括号,在格式化后仍需要将其显示。因此,您将需要使用一对大括号,用于格式设置的大括号,用于MathText功能的外部大括号。然后仍然需要转义外部的那些,以便不用于格式化。这导致3个花括号。

label = ("$y = {{{}}}x^{{{}}}$").format(round(popt[0],2), round(popt[1],2))

enter image description here