如果我已经有一个标题行,我希望在CSV文件中附加一个字典
如果一个值不存在,我想写'-999'
:
SDict ={T1:'A',T2:'B',T4:'D')
其中CSV文件的头为
T1,T2,T3,T4,T5
7,8,9,10,11
预期的结果是
T1,T2,T3,T4,T5
7,8,9,10,11
A,B,-999,D,-999
我正试着用代码:
import sys
import os
import csv
def GetFileHeader(Fpath):
i=10
ResFile=open (Fpath, 'r+')
HeaderDict={}
r=csv.reader(ResFile)
HeaderList = r.next()
for Header in HeaderList:
HeaderDict[Header]=i+1
print HeaderDict
ResFile.close()
return HeaderDict
Fpath= r'Z:\temp\assaf\S2TTP\S2T_TP\modules\results\Y124\res.csv'
Header= GetFileHeader(Fpath)
with open(Fpath,'rb') as fin:
dr = csv.DictReader(fin, dialect='excel')
print dr
print dr.fieldnames
# dr.fieldnames contains values from first row of `f`.
with open(Fpath,'ab+') as fou:
dw = csv.DictWriter(fou, dialect='excel', fieldnames=dr.fieldnames)
fieldnames=dr.fieldnames
for K in fieldnames:
dw.writerow(Header[k])
我想你可以简单地做:
这是假设您使用的是Python2.X。另外,请注意不要以换行结尾的文件,否则可能会以类似
7,8,9,10,11A,B,-999,D,-999
这样的行结尾。相关问题 更多 >
编程相关推荐