csvInput.py 文件源码

def readAdults(file_name):
    adult_file = open(file_name, 'r')
    adults_csv = csv.reader(adult_file, delimiter=',')

    # ignore the headers
    next(adults_csv, None)

    # create the dict we need
    adults = {}

    for adult_idx in adults_csv:
        adult = adults[int(adult_idx[0])] = {}
        adult['age'] = int(adult_idx[1])
        adult['workclass'] = adult_idx[2]
        adult['native-country'] = adult_idx[3]
        adult['sex'] = adult_idx[4]
        adult['race'] = adult_idx[5]
        adult['marital-status'] = adult_idx[6]

    adult_file.close()
    return adults