如何用線性回歸繪制散點圖？

Question

我如何使用matplotlib進行估算的行。

我有幾點，我使用以下代碼使用matplotlib進行了繪制：

import matplotlib.pyplot as plt
for smp, lbl in zip(samples, labels):
    plt.scatter(smp[0], smp[1], marker='*', cl = 'b', s=100, label=lbl)

# set limit, xlabel, ylabel, legend ...
# ...

plt.show()

謝謝，

Answer 1

使用polyfit進行線性回歸：

import matplotlib.pyplot as plt
from pylab import polyfit, poly1d

x, y = zip(*samples)

fit = polyfit(x, y, 1)
fit_fn = poly1d(fit)
plt.plot(x,y, '*', x, fit_fn(x), 'k')

plt.show()

結果示例：