在matplotlib上的绘图标签中显示数据

Question

有没有办法在我卓尔的情节中显示变量（特定的，线性拟合参数）？

另外，有没有办法控制显示的数字？

我正在使用matplotlib来生成数字，使用optimize.curve_fit来进行曲线拟合。

Answer 1

当然，只需使用text或annotate。

作为一个简单的例子：

import numpy as np
import matplotlib.pyplot as plt

# Generate data...
x = np.linspace(0, 30, 15)
y = 0.4 * x + 9 + np.random.random(x.size)

# Linear fit...
model = np.polyfit(x, y, deg=1)

plt.plot(x, y, 'bo', label='Observations')
plt.plot(x, model[0] * x + model[1], 'r-', label='Fitted Model')
plt.annotate(r'$y = {:.2f}x + {:.2f}$'.format(*model), xy=(0.1, 0.9), 
             xycoords='axes fraction', size=18)
plt.legend()

plt.show()