CODEWITHSUNDEEP
pythondictionaryreverse-lookup
Bows
Bows

Reputation: 49

Reverse lookup Dictionary

I have the following dictionary and I would like to do a reverse lookup.

countries_dict = {'Andorra': ('Europe', 'Andorra la Vella'),
              'Afghanistan': ('Asia', 'Kabul'),
              'Antigua and Barbuda': ('North America', "St. John's"),
              'Albania': ('Europe', 'Tirana'),
              'Armenia': ('Asia', 'Yerevan'),
              'Angola': ('Africa', 'Luanda'),
              'Argentina': ('South America', 'Buenos Aires'),
              'Austria': ('Europe', 'Vienna'),
              'Australia': ('Oceania', 'Canberra'),
              'Azerbaijan': ('Asia', 'Baku'),
              'Barbados': ('North America', 'Bridgetown'),
              'Bangladesh': ('Asia', 'Dhaka'),
              'Belgium': ('Europe', 'Brussels')}

My code looks like this:

def create_continents_dict(dct = dict) -> dict:
    reversed_dictionary = {value[0] : key for (key, value) in dct.items()}
    return reversed_dictionary

print(create_continents_dict(countries_dict))

{'Europe': 'Belgium', 'Asia': 'Bangladesh', 'North America': 'Barbados', 'Africa': 'Angola', 'South America': 'Argentina', 'Oceania': 'Australia'}

However I only get the last country it looks up as value. What I want is the value as a list of the countries. With the code above I should expect for every country that has the same continent, it should be in a list right?

{'Europe': ['Andorra', 'Albania', 'Austria', 'Belgium'] 'Asia': ['Afghanistan', 'Armenia', 'Azerbaijan', 'Bangladesh'], 'North America': ['Antigua and Barbuda', 'Barbados'], 'Africa': ['Angola'], 'South America': ['Argentina'], 'Oceania': ['Australia']}

Upvotes: 1

Views: 964

Answers (4)

chitown88
chitown88

Reputation: 28565

def create_continents_dict(dct = dict) -> dict:
    reversed_dictionary = {}
    for key, value in dct.items():
        if value[0] not in reversed_dictionary.keys():
            reversed_dictionary[value[0]] = []
        reversed_dictionary[value[0]].append(key)
    return reversed_dictionary

Output:

print(create_continents_dict(countries_dict))
{'Europe': ['Andorra', 'Albania', 'Austria', 'Belgium'], 'Asia': ['Afghanistan', 'Armenia', 'Azerbaijan', 'Bangladesh'], 'North America': ['Antigua and Barbuda', 'Barbados'], 'Africa': ['Angola'], 'South America': ['Argentina'], 'Oceania': ['Australia']}

Upvotes: 1

balderman
balderman

Reputation: 23815

Try

from collections import defaultdict 
countries_dict = {'Andorra': ('Europe', 'Andorra la Vella'),
              'Afghanistan': ('Asia', 'Kabul'),
              'Antigua and Barbuda': ('North America', "St. John's"),
              'Albania': ('Europe', 'Tirana'),
              'Armenia': ('Asia', 'Yerevan'),
              'Angola': ('Africa', 'Luanda'),
              'Argentina': ('South America', 'Buenos Aires'),
              'Austria': ('Europe', 'Vienna'),
              'Australia': ('Oceania', 'Canberra'),
              'Azerbaijan': ('Asia', 'Baku'),
              'Barbados': ('North America', 'Bridgetown'),
              'Bangladesh': ('Asia', 'Dhaka'),
              'Belgium': ('Europe', 'Brussels')}

d = defaultdict(list)
for k,v in countries_dict.items():
  d[v[0]].append(k)
print(d)

output

defaultdict(<class 'list'>, {'Europe': ['Andorra', 'Albania', 'Austria', 'Belgium'], 'Asia': ['Afghanistan', 'Armenia', 'Azerbaijan', 'Bangladesh'], 'North America': ['Antigua and Barbuda', 'Barbados'], 'Africa': ['Angola'], 'South America': ['Argentina'], 'Oceania': ['Australia']})

Upvotes: 1

Cory Kramer
Cory Kramer

Reputation: 117856

You could use collections.defaultdict here

def reverse_dict(countries_dict):
    reversed_dictionary  = defaultdict(list)
    for key, value in countries_dict.items():
        reversed_dictionary[value[0]].append(key)
    return reversed_dictionary

>>> reverse_dict(countries_dict)
defaultdict(<class 'list'>,
            {'Europe': ['Andorra', 'Albania', 'Austria', 'Belgium'],
             'Asia': ['Afghanistan', 'Armenia', 'Azerbaijan', 'Bangladesh'],
             'North America': ['Antigua and Barbuda', 'Barbados'],
             'Africa': ['Angola'],
             'South America': ['Argentina'],
             'Oceania': ['Australia']})

Upvotes: 1

Andrej Kesely
Andrej Kesely

Reputation: 195418

You can use dict.setdefault() with default list ([]) and then append the country in it:

out = {}
for country, (continent, _) in countries_dict.items():
    out.setdefault(continent, []).append(country)

print(out)

Prints:

{'Europe': ['Andorra', 'Albania', 'Austria', 'Belgium'], 
 'Asia': ['Afghanistan', 'Armenia', 'Azerbaijan', 'Bangladesh'], 
 'North America': ['Antigua and Barbuda', 'Barbados'], 
 'Africa': ['Angola'], 
 'South America': ['Argentina'], 
 'Oceania': ['Australia']}

Upvotes: 2

Related Questions