Python-使用CSV模块进行文件处理

来源:互联网 发布:python subprocess cwd 编辑:程序博客网 时间:2024/05/18 01:42

version 1:字段中还有逗号等复杂符号则无法处理

import osDATADIR = r"C:\Users\dell\Desktop\Udacity"DATAFILE = r"beatles-diskography.csv"def parseFile(path):    data = []    #f = open(path, "r")    with open(path, "rb") as f:        header = f.readline().split(",")        counter = 0        for line in f:            if counter == 10:                break            #print (line)            # return a list            fields = line.split(",")            #print(fields)            entry = {}            for index, eachvalue in enumerate(fields):                entry[header[index].strip()] = eachvalue.strip()            data.append(entry)            counter += 1    return datadef dealFile():    data = parseFile(os.path.join(DATADIR, DATAFILE))    print(data)dealFile()

version 2:使用csv模块

import csvimport osimport pprintDATADIR = r"C:\Users\dell\Desktop\Udacity"DATAFILE = r"beatles-diskography.csv"def parseFile(path):    data = []    n = 0    with open(path, "rb") as f:        r = csv.DictReader(f)        for line in r:            print line            data.append(line)    return dataif __name__ == '__main__':    path = os.path.join(DATADIR,DATAFILE)    result = parseFile(path)    print result    # format to output for read    pprint.pprint(result)

按照第一行为标签,处理成字典。

原创粉丝点击