2016-11-29 212 views
0

我有一个csv文件,其中包含一些行和列。我需要从该表中选取一些特定值,然后将其增强,然后将它们导出到另一个csv文件。导入csv,操作并导出到csv

我的Python 3代码如下所示:

import csv 
from urllib.parse import quote 

afn='https://r.refinedads.com/bs-mapping.php?v=a1&aid=7909&oid=1031&url=' 
zx='https://r.refinedads.com/bs-mapping.php?v=z1&aid=3814&oid=1031&zanpid=##zxClickID##&userid=##UserID##&url=' 
zxpn='https://r.refinedads.com/bs-mapping.php?v=z1&aid=3840&oid=1031&zanpid=##zxClickID##&userid=##UserID##&url=' 
la='https://r.refinedads.com/bs-mapping.php?aid=3843&oid=1031&c1=leadalliance&c2=XXXXX&c3=default&clid=YYYYY&url=' 
tv='https://r.refinedads.com/bs-mapping.php?aid=7639&oid=1031&c1=leadalliance&c2=XXXXX&c3=default&clid=YYYYY&url=' 


with open('input_csv.csv', 'r', newline='') as csvfile: 
    matrixreader = csv.DictReader(csvfile) 
    for row in matrixreader: 
     va = afn+quote(row['Link']+'&prf[pid]=[PARTNERID]&prf[a]=[AFFID]'+'-'+row['Promocode']) 
     vzx = zx+quote(row['Link']+'&prf[pid]=[PARTNERID]&prf[a]=[AFFID]'+'-'+row['Promocode']) 
     vzxpn = zxpn+quote(row['Link']+'&prf[pid]=[PARTNERID]&prf[a]=[AFFID]'+'-'+row['Promocode']) 
     vla = la+quote(row['Link']+'&prf[pid]=[PARTNERID]&prf[a]=[AFFID]'+'-'+row['Promocode']) 
     vtv = tv+quote(row['Link']+'&prf[pid]=[PARTNERID]&prf[a]=[AFFID]'+'-'+row['Promocode']) 
    liste = [[row['Name'], va, vzx, vzxpn, vla, vtv]] 
    print(liste) 

    with open('output.csv', 'w', newline='') as csvfile: 
     a = csv.writer(csvfile, delimiter=',') 
     a.writerows(liste) 

作为输出我刚刚收到的input.csv最后一排出口output.csv。在我看来,代码只是覆盖第一行。

input.csv看起来是这样的:

*Name,Promocode,Link 

BASE Plus + iPhone 7,ASDFNOWEDF,base.de/base-plus 

BASE Pro + iPhone 7,JBONEDGASD,base.de/base-pro 

BASE Light + iPhone 7,NAFODSFNTE,base.de/base-light 

BASE Pur + iPhone 7,NAEWRIONF,base.de/base-pur* 

非常感谢您在您的帮助, 干杯, 弗朗茨

+0

谢谢你们。 errikos是你的解决方案。 –

回答

0

尝试把如果循环内的清单当然分配 -

import csv 
from urllib.parse import quote 

afn='https://r.refinedads.com/bs-mapping.php?v=a1&aid=7909&oid=1031&url=' 
zx='https://r.refinedads.com/bs-mapping.php?v=z1&aid=3814&oid=1031&zanpid=##zxClickID##&userid=##UserID##&url=' 
zxpn='https://r.refinedads.com/bs-mapping.php?v=z1&aid=3840&oid=1031&zanpid=##zxClickID##&userid=##UserID##&url=' 
la='https://r.refinedads.com/bs-mapping.php?aid=3843&oid=1031&c1=leadalliance&c2=XXXXX&c3=default&clid=YYYYY&url=' 
tv='https://r.refinedads.com/bs-mapping.php?aid=7639&oid=1031&c1=leadalliance&c2=XXXXX&c3=default&clid=YYYYY&url=' 


with open('input_csv.csv', 'r', newline='') as csvfile: 
    matrixreader = csv.DictReader(csvfile) 
    liste=[] # initialize here 
    for row in matrixreader: 
     va = afn+quote(row['Link']+'&prf[pid]=[PARTNERID]&prf[a]=[AFFID]'+'-'+row['Promocode']) 
     vzx = zx+quote(row['Link']+'&prf[pid]=[PARTNERID]&prf[a]=[AFFID]'+'-'+row['Promocode']) 
     vzxpn = zxpn+quote(row['Link']+'&prf[pid]=[PARTNERID]&prf[a]=[AFFID]'+'-'+row['Promocode']) 
     vla = la+quote(row['Link']+'&prf[pid]=[PARTNERID]&prf[a]=[AFFID]'+'-'+row['Promocode']) 
     vtv = tv+quote(row['Link']+'&prf[pid]=[PARTNERID]&prf[a]=[AFFID]'+'-'+row['Promocode']) 
     liste = liste.append([[row['Name'], va, vzx, vzxpn, vla, vtv]]) # keep appending 
    print(liste) 

    with open('output.csv', 'w', newline='') as csvfile: 
     a = csv.writer(csvfile, delimiter=',') 
     a.writerows(liste) 
1

一个问题是,您尝试在读取循环之外更新liste变量。另一个问题是你重新分配给liste变量,而不是appending

但是,您不需要收集列表中的所有行,然后将它们写入输出。您可以通过在阅读并形成时刻写每行来做到这一点。

with open('input_csv.csv', 'r', newline='') as csvfile_in: 
    with open('output.csv', 'w', newline='') as csvfile_out: 
     matrixreader = csv.DictReader(csvfile_in) 
     a = csv.writer(csvfile_out, delimiter=',') 
     for row in matrixreader: 
      va = afn+quote(row['Link']+'&prf[pid]=[PARTNERID]&prf[a]=[AFFID]'+'-'+row['Promocode']) 
      vzx = zx+quote(row['Link']+'&prf[pid]=[PARTNERID]&prf[a]=[AFFID]'+'-'+row['Promocode']) 
      vzxpn = zxpn+quote(row['Link']+'&prf[pid]=[PARTNERID]&prf[a]=[AFFID]'+'-'+row['Promocode']) 
      vla = la+quote(row['Link']+'&prf[pid]=[PARTNERID]&prf[a]=[AFFID]'+'-'+row['Promocode']) 
      vtv = tv+quote(row['Link']+'&prf[pid]=[PARTNERID]&prf[a]=[AFFID]'+'-'+row['Promocode']) 
      a.writerow([row['Name'], va, vzx, vzxpn, vla, vtv]) 

这样更有效率和内存友好,特别是如果输入文件非常大。