#!/usr/bin/env python

from __future__ import division

import numpy as np
from matplotlib import pyplot as plt



def main():
    koeffs = [.3, 1.2, .1, 7]
    p = np.poly1d(koeffs)
    x = np.linspace(-2, 2, 100)
    y = p(x) + 2 * np.random.randn(100) - 1
    # fit
    fit =  np.polyfit(x, y, 3)
    p_fit = np.poly1d(fit)
    # plot
    plt.scatter(x, y)
    plt.plot(x, p_fit(x))
    plt.show()

if __name__ == '__main__':
    main()

#!/usr/bin/env python

from __future__ import division

import numpy as np
from matplotlib import pyplot as plt



def main():
    koeffs = [.3, 1.2, .1, 7]
    p = np.poly1d(koeffs)
    x = np.linspace(-2, 2, 100)
    y = p(x) + 2 * np.random.randn(100) - 1
    # fit
    fit =  np.polyfit(x, y, 3)
    p_fit = np.poly1d(fit)
    print "Real coefficients:", koeffs
    print "Fitted coefficients:", fit
    # plot
    plt.scatter(x, y)
    plt.plot(x, p_fit(x))
    plt.show()

if __name__ == '__main__':
    main()

Print real and fitted coeffs
Print real and fitted coeffs
