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

Question

有没有办法在我绘制的图中显示变量(特别是线性拟合参数)？

还有，有没有办法控制显示的数字数量？

我使用 matplotlib 生成图形，并使用 optimize.curve_fit 进行曲线拟合。

Answer 1

当然，只需使用文本或注释即可。

举个简单的例子:

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