python 写CSV

Question

问题相关代码

for i in range(0,len(link_start)):
    trajOD,etaOD = traj_judge(link_start[i],link_start5,link_end[i],link_end5,eta)
    print('trajOD:',trajOD ,'\n','etaOD:',etaOD)
    if len(trajOD) == 1:
        etaODAVG = simple_eta[i]
    else:
        etaODAVG = etaOD.mean()
    #etaODAVG = etaOD.mean()
    print('trajOD:',trajOD,'\n','trajODlen:', len(trajOD),'\n','etaODAVG:',etaODAVG,'eat:',eta[i])

我想要达到的结果

我想把每次循环输出的结果写入一个CSV文件，请问该怎么写代码呀？

Answer 1

file = open('whatever.csv', 'w')
for i in range(0, len(link_start)):
    trajOD, etaOD = traj_judge(link_start[i], link_start5, link_end[i], link_end5, eta)
    print('trajOD:', trajOD, '\n', 'etaOD:', etaOD)
    if len(trajOD) == 1:
        etaODAVG = simple_eta[i]
    else:
        etaODAVG = etaOD.mean()
    # etaODAVG = etaOD.mean()
    print('trajOD:', trajOD, '\n', 'trajODlen:', len(trajOD), '\n', 'etaODAVG:', etaODAVG, 'eat:', eta[i])
    file.write(",".join((trajOD, len(trajOD), etaODAVG, eta[i])) + "\n")
file.close()