polyfit_plot.py

import numpy as np
from matplotlib import pyplot as plt

def polyfit_plot(x, y, p=1):
    plt.scatter(x, y)
    axes = plt.gca()
    coeff = np.polyfit(x, y, p)
    X_plot = np.linspace(axes.get_xlim()[0],axes.get_xlim()[1],100)
    print (np.corrcoef(x,y)[0,1])
    Y_plot = 0
    for c in coeff:
        Y_plot = X_plot*Y_plot + c
    plt.plot(X_plot, Y_plot, '-')