Python 101 Assignment 3 Solution 1

solution_1.py

people = {} # let's start with an empty dictionary to hold our data

with open('numbers.txt', 'r') as numbers:
    for index, line in enumerate(numbers):
        if index > 0: # skip the first line (don't need the headers)
            name, birthday = line.rstrip().split(',') 

            # study this line carefully, it's intentionally terse
            people.update({ name : { 'number' : birthday } })


with open('birthdays.txt', 'r') as birthdays:
    for line in birthdays:
        name, birthday = line.rstrip().split(',')

        try:
            people[name]['birthday'] = birthday # Update each person in the dictionary with the birthday data
        except KeyError, e: # skip any missing keys i.e. the header column
            print "KeyError: Skip this line, {} not found".format(e)


with open('output.txt', 'w') as output:
    
    # Write column headers
    output.write('name,birthday,number\n')

    # Write each line to the new output.txt file
    for key, value in people.iteritems():
        name, number, birthday = key, value['number'], value['birthday']
        line = '{},{},{}\n'.format(name, number, birthday)
        output.write(line)