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

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

import os

DATADIR = 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 data

def dealFile():
    data = parseFile(os.path.join(DATADIR, DATAFILE))
    print(data)

dealFile()

version 2：使用csv模块

import csv
import os
import pprint

DATADIR = 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 data

if __name__ == '__main__':
    path = os.path.join(DATADIR,DATAFILE)
    result = parseFile(path)
    print result

    # format to output for read
    pprint.pprint(result)

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