file = open(outputfile,'a')
    #f.writelines()每行写入多个变量
    #f.write()每行写入单个变量
    for b in range(index_540,index_840+1):
        for i in range(1,8):
            I_lambda_0 = []
            I_lambda_alpha = []
            alpha = []
            for f in range(len(fns)):
                df = pd.read_csv(os.path.join(input_dir,fns[f]))
                I_lambda_0.append(df.loc[0,cols[b]])
                I_lambda_alpha.append(df.loc[i,cols[b]])
                alpha.append(i*10/180*np.pi)  #转换为弧度制
            I_lambda_alpha = np.array(I_lambda_alpha)
            I_lambda_0 = np.array(I_lambda_0)
            alpha = np.array(alpha)
            parameters = leastsq_func(residuals,p0,I_lambda_alpha,I_lambda_0,alpha)
            #parameters[0]为拟合后的参数
            k = np.array(parameters[0][0])
            m = np.array(parameters[0][1])
            #file.writelines('\n'+cols[b] + ' ' +str(i*10) +' ' + str(k)+' '+str(m))
            #上述写入也可以
            file.write('{:} {:} {:} {:}\n'.format(cols[b],str(i*10),k,m))#格式化写入也可以
    file.close()

本文地址:https://blog.csdn.net/qq_37970770/article/details/110881579