birthdays_by_month_using_dictionaries.py

#  TODO Make the month_names list from the months module available
from months import month_names

#  TODO Make the sort_by_month function from the months module 
#  available as sort_by_month_name

#  Make the matplotlib module available
import matplotlib

#  Set up rendering of matplotlib so that it doesn't try to
#  display the plot
matplotlib.use('Agg')

#  Make the plotting functions of matplotlib available
import matplotlib.pyplot as plt

#  Ask the user for how many people are in the class, storing the
#  number entered into an integer variable named number_of_students
number_of_students = int(raw_input('How many people are in the class? '))

#  Initialize a variable named number_entered to keep of how many people we have entered
number_entered = 0

#  Initialize an empty dictionary named month_counts
month_counts = dict()
people_by_month = dict()

#  TODO Set the value associated with each month in months_names to 0 in the
#  dictionary named month_counts
for month in month_names:
    month_counts[month] = 0
    people_by_month[month] = list()

#  As long as there are more people
#  TODO:  Fill in a comparison of number_entered with number_of_students
while number_entered < number_of_students:
  #  Ask for the next person's birthdate
  line = raw_input('Enter name and birthday, separated by comma: ')

  #  TODO Split up the line into parts, separated by the , character, storing the 
  #  result into a list named name_and_birthday
  name_and_birthday = line.split(',')
  print "The name is ", name_and_birthday[0], 'and birthday is', name_and_birthday[1]

  #  TODO Strip off any spaces from the birthday portion of name_and_birthday, then 
  #  split based on a space, storing the two parts into a list named 
  #  month_and_day
  month_and_day = name_and_birthday[1].strip(' ').split(' ')

  #  TODO Set the variable "month" to be the 3 letter abbreivation, which is
  #  one of the elements in the list named month_and_day
  month = month_and_day[0]

  #  TODO Increment the number of people whose birthdays we've entered, which
  #  is in the variable number_entered
  number_entered += 1

  #  TODO Increment the value in the month_count dictionary for this month
  month_counts[month] += 1
  people_by_month[month].append(name_and_birthday[0]) 

#  Get out the x_values, which are the month names
x_values = month_names

#  Create a list of the y-values from the month_counts dictionary, in the same order as the values in the months_names
#  list
y_values = list()
for month in month_names:
    y_values.append(month_counts[month])

#  Create a plot object with formatting specifying no line, 
#  just circles, for the values in x_values and y_values
plt.plot(x_values, y_values, 'ro')

#  Save that plot to the file months-dictionary.png
plt.savefig('months-dictionary-wednesday-3.png')

for month in month_names:
    if len(people_by_month[month]) > 0:
        print month, people_by_month[month]
    else:
        print month, ": no one born in", month