отображение данных в метке графика Matplot lib

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