switch key and values in a dict of lists

Question

Hello Stackoverflow people,

I have a nested dictionary with lists as values and I want to create a dict where all the list entries get their corresponding key as value.

Example time!

# what I have
dict1 = {"A":[1,2,3], "B":[4,5,6], "C":[7,8,9]}

# what I want
dict2 = {1:"A", 2:"A", 3:"A", 4:"B", 5:"B", 6:"B", 7:"C", 8:"C", 9:"C"}

Any help will be much appreciated!

What happens if you have duplicate values? For example: `{"A":[1, 2], "B":[2, 3]}` ? — Dani Mesejo, Jan 29 '19 at 13:40

score 3 · Accepted Answer · answered Jan 29 '19 at 13:32

3

Try this

dict1 = {"A":[1,2,3], "B":[4,5,6], "C":[7,8,9]}
dict2= {}
for keys,values in dict1.items():
    for i in values:
        dict2[i]=keys
print(dict2)

Output

{1: 'A', 2: 'A', 3: 'A', 4: 'B', 5: 'B', 6: 'B', 7: 'C', 8: 'C', 9: 'C'}

Hope it helps

answered Jan 29 '19 at 13:32

Talha Israr

654
6
17

Well that was easy! Thanks a lot! – the_man_in_black Jan 29 '19 at 13:57

aquil.abdullah · Answer 2 · 2019-01-29T14:01:40.303

2

Use dictionary comprehension:

d = {'a': 'b', 'c': 'd', 'e': 'f'}
d2 = dict((v1, k) for k, v in d.items() for v1 in v)  # Here is the one-liner

edited Jan 29 '19 at 14:01

answered Jan 29 '19 at 13:51

aquil.abdullah

3,059
3
21
40

1

It should be noted that the solution above only works if the values are unique. – aquil.abdullah Jan 29 '19 at 13:58

score 1 · Answer 3 · answered Jan 29 '19 at 14:06

1

assuming your key: value dictionary contains list as a value and using dict comprehension.

Using a second loop to iterate over the list present in original dictionary.

{item: key for key, value in dict1.items() for item in value}

answered Jan 29 '19 at 14:06

Vishal Singh

6,014
2
17
33

switch key and values in a dict of lists

3 Answers3

Linked