# File: Exercise2_4.py (Chapter 2, Exercise 4)
# Purpose: Calculate Average drug use per year for each drug
# Author: Emily G. Allen

# Import pylab
from pylab import *

# Import drug data
from DrugData import *

def calcAverages():

    # calculate the average (mean) of each drug type
    aMarijuana = mean(Marijuana)
    aHallucinogens = mean(Hallucinogens)
    aCocaine = mean(Cocaine)
    aHeroine = mean(Heroin)
    aCigarettes = mean(Cigarettes)
    aAlcohol = mean(Alcohol)

    # save all the averages into an array
    averageUse = [aMarijuana, aHallucinogens, aCocaine, aHeroine, aCigarettes, aAlcohol]
    
    return averageUse # return the list of values stored in the array