CODEWITHSUNDEEP
python
NeoSoul
NeoSoul

Reputation: 122

Merge multiple key in dictionary with same values inside an array

I need to merge three dictionaries with array values into a single dict based on if they have same values inside the array. The dictionaries is like this:

data = {
'A1':['Cheese','Cupcake', 'Salad','Sandwich'],
'A2':['Cheese','Cupcake', 'Pasta','Pudding'],
'A3':['Pudding','Pasta', 'Salad','Sandwich']
}

Then, the output would be like this:

{
   'A1,A2':['Cheese','Cupcake']
   'A1,A3':['Salad', 'Sandwich']
   'A2,A3':['Pudding','Pasta']
}

I've tried this:

tmp = {}
for key, value in data.items():
  if value in tmp:
    tmp[value].append(key)
  else:
    tmp[value] = [ key ]
print(tmp)

But it only works if the values isn't a list or array. Any solution?

Upvotes: 0

Views: 66

Answers (1)

mozway
mozway

Reputation: 260825

Given you use case, you could use itertools.combinations and set intersection:

data = {
'A1':['Cheese','Cupcake', 'Salad','Sandwich'],
'A2':['Cheese','Cupcake', 'Pasta','Pudding'],
'A3':['Pudding','Pasta', 'Salad','Sandwich']
}


from itertools import combinations

out = {f'{a},{b}': list(set(data[a]).intersection(data[b]))
       for a,b in combinations(data, 2)}

output:

{'A1,A2': ['Cupcake', 'Cheese'],
 'A1,A3': ['Sandwich', 'Salad'],
 'A2,A3': ['Pasta', 'Pudding']}

Upvotes: 3

Related Questions