TerritoryCode.py

Go to the documentation of this file.

 
territory_iso_code = {'Andorra': 'AD', 'The United Arab Emirates': 'AE', 'Afghanistan': 'AF', 'Antigua and Barbuda': 'AG', 'Anguilla': 'AI', 'Albania': 'AL', 
'Armenia': 'AM', 'Netherlands Antilles': 'AN', 'Angola': 'AO', 'Antarctica': 'AQ', 'Argentina': 'AR', 'American Samoa': 'AS', 'Austria': 'AT', 'Australia': 'AU', 
'Aruba': 'AW', 'Åland Islands': 'AX', 'Azerbaijan': 'AZ', 'Bosnia and Herzegovina': 'BA', 'Barbados': 'BB', 'Bangladesh': 'BD', 'Belgium': 'BE', 
'Burkina Faso': 'BF', 'Bulgaria': 'BG', 'Bahrain': 'BH', 'Burundi': 'BI', 'Benin': 'BJ', 'Saint Barthélemy': 'BL', 'Bermuda': 'BM', 'Brunei': 'BN', 
'Bolivia': 'BO', 'Bonaire, Sint Eustatius and Saba': 'BQ', 'Brazil': 'BR', 'The Bahamas': 'BS', 'Bhutan': 'BT', 'Bouvet Island': 'BV', 'Botswana': 'BW', 
'Belarus': 'BY', 'Belize': 'BZ', 'Canada': 'CA', 'Cocos Islands': 'CC', 'The Democratic Republic of the Congo': 'CD', 'The Central African Republic': 'CF', 
'The Congo': 'CG', 'Switzerland': 'CH', "Côte d'Ivoire": 'CI', 'Cook Islands': 'CK', 'Chile': 'CL', 'Cameroon': 'CM', 'China': 'CN', 'Colombia': 'CO', 
'Costa Rica': 'CR', 'Serbia and Montenegro': 'CS', 'Cuba': 'CU', 'Cape Verde': 'CV', 'Curaçao': 'CW', 'Christmas Island': 'CX', 'Cyprus': 'CY', 'Czechia': 'CZ', 
'Germany': 'DE', 'Djibouti': 'DJ', 'Denmark': 'DK', 'Dominica': 'DM', 'The Dominican Republic': 'DO', 'Algeria': 'DZ', 'Ecuador': 'EC', 'Estonia': 'EE', 'Egypt': 
'EG', 'Western Sahara': 'EH', 'Eritrea': 'ER', 'Spain': 'ES', 'Ceuta, Spain': 'ES-CE', 'Canary Islands, Spain': 'ES-CN', 'Melilla, Spain': 'ES-ML', 'Ethiopia': 'ET', 
'Finland': 'FI', 'Fiji': 'FJ', 'Falkland Islands': 'FK', 'Micronesia': 'FM', 'Faroe Islands': 'FO', 'France': 'FR', 'Gabon': 'GA', 'The United Kingdom': 'GB', 
'Grenada': 'GD', 'Georgia': 'GE', 'French Guiana': 'GF', 'Guernsey': 'GG', 'Ghana': 'GH', 'Gibraltar': 'GI', 'Greenland': 'GL', 'The Gambia': 'GM', 'Guinea': 'GN', 
'Guadeloupe': 'GP', 'Equatorial Guinea': 'GQ', 'Greece': 'GR', 'South Georgia and The South Sandwich Islands': 'GS', 'Guatemala': 'GT', 'Guam': 'GU', 
'Guinea-Bissau': 'GW', 'Guyana': 'GY', 'Hong Kong': 'HK', 'Heard Island and McDonald Islands': 'HM', 'Honduras': 'HN', 'Croatia': 'HR', 'Haiti': 'HT', 
'Hungary': 'HU', 'Indonesia': 'ID', 'Ireland': 'IE', 'Israel': 'IL', 'Isle of Man': 'IM', 'India': 'IN', 'The British Indian Ocean Territory': 'IO', 'Iraq': 'IQ', 
'Iran': 'IR', 'Iceland': 'IS', 'Italy': 'IT', 'Jersey': 'JE', 'Jamaica': 'JM', 'Jordan': 'JO', 'Japan': 'JP', 'Kenya': 'KE', 'Kyrgyzstan': 'KG', 'Cambodia': 'KH', 
'Kiribati': 'KI', 'The Comoros': 'KM', 'Saint Kitts and Nevis': 'KN', 'Korea (North)': 'KP', 'Korea (South)': 'KR', 'Kuwait': 'KW', 'Cayman Islands': 'KY', 
'Kazakhstan': 'KZ', 'Laos': 'LA', 'Lebanon': 'LB', 'Saint Lucia': 'LC', 'Liechtenstein': 'LI', 'Sri Lanka': 'LK', 'Liberia': 'LR', 'Lesotho': 'LS', 
'Lithuania': 'LT', 'Luxembourg': 'LU', 'Latvia': 'LV', 'Libya': 'LY', 'Morocco': 'MA', 'Monaco': 'MC', 'Moldova': 'MD', 'Montenegro': 'ME', 'Saint Martin': 'MF', 
'Madagascar': 'MG', 'The Marshall Islands': 'MH', 'Macedonia': 'MK', 'Mali': 'ML', 'Myanmar': 'MM', 'Mongolia': 'MN', 'Macao': 'MO', 'Northern Mariana Islands': 'MP', 
'Martinique': 'MQ', 'Mauritania': 'MR', 'Montserrat': 'MS', 'Malta': 'MT', 'Mauritius': 'MU', 'Maldives': 'MV', 'Malawi': 'MW', 'Mexico': 'MX', 'Malaysia': 'MY', 
'Mozambique': 'MZ', 'Namibia': 'NA', 'New Caledonia': 'NC', 'The Niger': 'NE', 'Norfolk Island': 'NF', 'Nigeria': 'NG', 'Nicaragua': 'NI', 'The Netherlands': 'NL', 
'Norway': 'NO', 'Nepal': 'NP', 'Nauru': 'NR', 'Niue': 'NU', 'New Zealand': 'NZ', 'Oman': 'OM', 'Panama': 'PA', 'Peru': 'PE', 'French Polynesia': 'PF', 
'Papua New Guinea': 'PG', 'The Philippines': 'PH', 'Pakistan': 'PK', 'Poland': 'PL', 'Saint Pierre and Miquelon': 'PM', 'Pitcairn': 'PN', 'Puerto Rico': 'PR', 
'The Occupied Palestinian Territory': 'PS', 'Portugal': 'PT', 'Palau': 'PW', 'Paraguay': 'PY', 'Qatar': 'QA', 'Réunion': 'RE', 'Romania': 'RO', 'Serbia': 'RS', 
'Russia': 'RU', 'Rwanda': 'RW', 'Saudi Arabia': 'SA', 'Solomon Islands': 'SB', 'Seychelles': 'SC', 'The Sudan': 'SD', 'Sweden': 'SE', 'Singapore': 'SG', 
'Saint Helena': 'SH', 'Slovenia': 'SI', 'Svalbard and Jan Mayen': 'SJ', 'Slovakia': 'SK', 'Sierra Leone': 'SL', 'San Marino': 'SM', 'Senegal': 'SN', 'Somalia': 'SO', 
'Suriname': 'SR', 'South Soudan': 'SS', 'Sao Tome and Principe': 'ST', 'El Salvador': 'SV', 'Sint Maarten': 'SX', 'Syria': 'SY', 'Eswatini': 'SZ', 
'Turks and Caicos Islands': 'TC', 'Chad': 'TD', 'The French Southern Territories': 'TF', 'Togo': 'TG', 'Thailand': 'TH', 'Tajikistan': 'TJ', 'Tokelau': 'TK', 
'Timor-Leste': 'TL', 'Turkmenistan': 'TM', 'Tunisia': 'TN', 'Tonga': 'TO', 'Turkey': 'TR', 'Trinidad and Tobago': 'TT', 'Tuvalu': 'TV', 'Taiwan': 'TW', 
'Tanzania': 'TZ', 'Ukraine': 'UA', 'Uganda': 'UG', 'United States Minor Outlying Islands': 'UM', 'The United States': 'US', 'Uruguay': 'UY', 'Uzbekistan': 'UZ', 
'The Holy See': 'VA', 'Saint Vincent and The Grenadines': 'VC', 'Venezuela': 'VE', 'British Virgin Islands': 'VG', 'US Virgin Islands': 'VI', 'Viet Nam': 'VN', 
'Vanuatu': 'VU', 'Wallis and Futuna': 'WF', 'Samoa': 'WS', 'Yemen': 'YE', 'Mayotte': 'YT', 'South Africa': 'ZA', 'Zambia': 'ZM', 'Zimbabwe': 'ZW', 'Bahamas': 'BS', 
'Bolivia (Plurinational State of)': 'BO', 'Bosnia-Hercegovina': 'BA', 'British Indian Ocean Territory': 'IO', 'Brunei Darussalam': 'BN', 'Cabo Verde': 'CV', 
'Central African Republic': 'CF', 'China (PRC)': 'CN', 'Cocos (Keeling) Islands': 'CC', 'Comoros': 'KM', 'Congo': 'CG', 'Congo, Democratic Republic of the': 'CD', 
'Czech Republic': 'CZ', 'Dominican Republic': 'DO', 'Falkland Islands (Malvinas)': 'FK', 'French Southern Territories': 'TF', 'Gambia': 'GM', 'Holy See': 'VA', 
'Iran (Islamic Republic of)': 'IR', 'Kyrgyz Republic': 'KG', "Lao People's Democratic Republic": 'LA', 'North Macedonia': 'MK', 'Marshall Islands': 'MH', 
'Netherlands': 'NL', 'Niger': 'NE', 'Palestine, State of': 'PS', 'Philippines': 'PH', 'Saint Helena, Ascension and Tristan da Cunha': 'SH', 
'Saint Martin (French part)': 'MF', 'Saint Vincent and the Grenadines': 'VC', 'Sint Maarten (Dutch part)': 'SX', 'Slovak Republic': 'SK', 
'South Georgia and the South Sandwich Islands': 'GS', 'South Sudan': 'SS', 'Sudan': 'SD', 'Syrian Arab Republic': 'SY', 'Tanzania, United Republic of': 'TZ', 
'USA': 'US', 'United Arab Emirates': 'AE', 'United Kingdom': 'GB', 'United States of America': 'US', 'Vietnam': 'VN', 'Virgin Islands (British)': 'VG', 
'Virgin Islands (U.S.)': 'VI', 'Zanzibar': 'TZ'}
 
class TerritoryCode:
    """
    Maps the full country name to ISO-2 code
    """
    def terr_iso(self, country, iso):
        """Takes each country name and searches their ISO-2 code in the territory_iso_code
 
        Args:
            country (str): current country
            iso (dict): ISO codes for countries
        Returns: ISO-2 code
        """
        return iso[country]
 
    def territory_code(self, df, territory_iso_code=territory_iso_code):
        """Applies terr_iso for all territory column
 
        Args:
            df (pandas dataframe): current file loaded as dataframe
            territory_iso_code (dict): ISO codes for countries
        Returns: df (pandas dataframe)
        """
        df["Territory"] = df.apply(lambda Row: self.terr_iso(Row['Territory'], territory_iso_code), axis=1)
        return df