def test_country_name_3(self): """ Check whether country name was correctly renamed pre-merge""" data = pd.DataFrame(['Russian Federation', 'Macedonia', 'Burkina'], columns=['Country']) self.assertEqual(list(data.Country[data.Country == 'Burkina'].values), ['Burkina']) self.assertEqual( list( change.change_name(data).Country[ data.Country == 'Burkina'].values), ['Burkina Faso'])
def merge_by_year(year): """Merges 'countries' with 'income' using 'Country' as key """ """This is an inner join because we cannot use un-matched countries in plots """ income_data = pd.DataFrame(income[year]) income_data = income_data.rename(columns = {year:'Income'}) # ! Important! # I notice that sometimes country names differ in the two datatest and hence change 14 names in 'countries' data to ensure a match. See change_country_name.py for more information. Inner join n goes up from 177 to 191. countries_renamed = change.change_name(countries) # Country is a column in 'countries_renamed' but index in income_data merged_data =pd.merge(countries_renamed, income_data, left_on = 'Country',right_index=True) return merged_data
def merge_by_year(year): """Merges 'countries' with 'income' using 'Country' as key """ """This is an inner join because we cannot use un-matched countries in plots """ income_data = pd.DataFrame(income[year]) income_data = income_data.rename(columns={year: 'Income'}) # ! Important! # I notice that sometimes country names differ in the two datatest and hence change 14 names in 'countries' data to ensure a match. See change_country_name.py for more information. Inner join n goes up from 177 to 191. countries_renamed = change.change_name(countries) # Country is a column in 'countries_renamed' but index in income_data merged_data = pd.merge(countries_renamed, income_data, left_on='Country', right_index=True) return merged_data
def test_country_name_3(self): """ Check whether country name was correctly renamed pre-merge""" data = pd.DataFrame(['Russian Federation', 'Macedonia', 'Burkina'], columns=['Country']) self.assertEqual(list(data.Country[data.Country=='Burkina'].values),['Burkina']) self.assertEqual(list(change.change_name(data).Country[data.Country=='Burkina'].values),['Burkina Faso'])