最小二乘法拟合方位角与径向速度的关系

# -*- coding: GB2312 -*-
"""
Created on Thu Jan 21 20:56:24 2016

@author: Administrator
"""

import numpy as np
from scipy.optimize import leastsq
import pylab as pl
import sys

def func(x, p):
    """
    数据拟合所用的函数: a + b*sin(cx + d)
    """
    a, b, c, d = p
    return a + b*np.sin(c*x+d)
 
def residuals(p, y, x):
    """
    实验数据x, y和拟合函数之间的差，p为拟合需要找到的系数
    """
    return y - func(x, p)
 

file=open("E:\\fit\\AM_new.csv",'r')
dataLines = file.readlines()
angle = []
dataA = []
#dataB = []
#dataC = []
#dataD = []

for line in dataLines:
    angleCur,dataACur,dataBCur,dataCCur,dataDCur = line.split(',')
    
    if angleCur != 'NaN':
        angle.append(float(angleCur)/180*np.pi)
    if dataACur != 'NaN':