Exemplo n.º 1
0
def merge_by_year(year):
    '''
    The required funtion for question 5
    '''
    income = transformIncome()
    countries = loadCountries()
    
    # get the income of a particular year
    income_year = income.loc[[year]]
    #transpose the dataframe back
    income_year = income_year.T
    income_year.index.name = 'Country'
    merged = countries.join(income_year, on = 'Country', how = 'right')
    merged.rename(columns = {year: 'Income'}, inplace = True)
    return merged
Exemplo n.º 2
0
def checkYear(input):
    '''
    check if the input year exists in the data
    '''
    income = transformIncome()
    
    # create a list of years appearing in the data
    year_list = list(income.index)
    
    if input.isdigit():
        year = int(input)
        if year in year_list:
            return True
        else:
            return False
    else:
        return False
Exemplo n.º 3
0
def plotIncome(input):
    '''
    given a valid input, plot the distribution of the income
    '''
    income = transformIncome()
    
    title = 'Distribution of income per capita in ' + input
    
    # columns not included in our plot
    year = int(input)
    null_idx = income.loc[year].isnull()
    values = income.loc[year, ~null_idx].values
    plt.figure()
    
    # plot the histogram
    plt.hist(values, bins = 20)
    plt.title(title)
    plt.ylabel('Frequency')
    plt.xlabel('Income per capita')
    plt.show()