How to search a nested dictionary for a given value

user_13806407 0
Viewed 85

I have a dictionary that looks like this:

{'Week 1': {'Game1': {'Away': 'Team3', 'Home': 'Team2', 'Venue': 5},
            'Game2': {'Away': 'Team1', 'Home': 'Team4', 'Venue': 6}},
 'Week 2': {'Game1': {'Away': 'Team1', 'Home': 'Team2', 'Venue': 4},
            'Game2': {'Away': 'Team4', 'Home': 'Team3', 'Venue': 1}},
 'Week 3': {'Game1': {'Away': 'Team1', 'Home': 'Team3', 'Venue': 6},
            'Game2': {'Away': 'Team2', 'Home': 'Team4', 'Venue': 4}} 
...}

The dictionary contains fixture seperated by 'Week', 'Game' then 'Home', 'Away' and 'Venue'.

Say I was looking to search for all fixtures for 'Team 1' from the dictionary above and get an output of:

{'Week 1': {'Game 2': {'Away: ': 'a', 'Home: ': 'd', 'Rink ': 6}},
 'Week 2': {'Game1': {'Away: ': 'a', 'Home: ': 'b', 'Rink ': 4}},
 'Week 3': {'Game1': {'Away: ': 'a', 'Home: ': 'c', 'Rink ': 6}}}

What would be the best way to do this?

I've tried using for loops

for week, games in dict:
    for game, details in games:
        for k, v in details:
            if v == 'Team1'
               newDict.update({week: {game: details}})

But i get ValueError: too many values to unpack

Am I on the right lines, or totally off base? Any help would be greatly appreciated

dictionary nested python
3 Answers

Dict Comprehension:

{match: {game: details} for match, games in a.items() for game, details in games.items() for k, v in details.items() if v == 'Team1'}
user_100420320

You can also just do:

{
  week: {game: info}
  for week, games in weeks
  for game, info in games
  if team in (info['Away'], info['Home'])
}

Putting it all together:

>>> weels = {'Week 1': {'Game1': {'Away': 'Team3', 'Home': 'Team2', 'Venue': 5},
...             'Game2': {'Away': 'Team1', 'Home': 'Team4', 'Venue': 6}},
...  'Week 2': {'Game1': {'Away': 'Team1', 'Home': 'Team2', 'Venue': 4},
...             'Game2': {'Away': 'Team4', 'Home': 'Team3', 'Venue': 1}},
...  'Week 3': {'Game1': {'Away': 'Team1', 'Home': 'Team3', 'Venue': 6},
...             'Game2': {'Away': 'Team2', 'Home': 'Team4', 'Venue': 4}} }
>>> team = 'Team1'
>>> {week: {game: info}
...   for week, games in weeks
...   for game, info in games
...   if team in (info['Away'], info['Home'])}
{'Week 1': {'Game2': {'Away': 'Team1', 'Home': 'Team4', 'Venue': 6}}, 'Week 2': {'Game1': {'Away': 'Team1', 'Home': 'Team2', 'Venue': 4}}, 'Week 3': {'Game1': {'Away': 'Team1', 'Home': 'Team3', 'Venue': 6}}}
user_21536360

You can use a NestedDict and avoid using nested for loops or comprehensions.

from ndicts.ndicts import NestedDict

d = {'Week 1': {'Game1': {'Away': 'Team3', 'Home': 'Team2', 'Venue': 5},
                'Game2': {'Away': 'Team1', 'Home': 'Team4', 'Venue': 6}},
     'Week 2': {'Game1': {'Away': 'Team1', 'Home': 'Team2', 'Venue': 4},
                'Game2': {'Away': 'Team4', 'Home': 'Team3', 'Venue': 1}},
     'Week 3': {'Game1': {'Away': 'Team1', 'Home': 'Team3', 'Venue': 6},
                'Game2': {'Away': 'Team2', 'Home': 'Team4', 'Venue': 4}}}
nd = NestedDict()

nd_team1 = {}
for key, value in nd.items():
    if value == "Team1":
            team1_record = nd.extract[key[:-1]]
            nd_team1.update(team1_record)

>>> nd_team1.to_dict()
{'Week 1': {'Game2': {'Away': 'Team1', 'Home': 'Team4', 'Venue': 6}},
 'Week 2': {'Game1': {'Away': 'Team1', 'Home': 'Team2', 'Venue': 4}},
 'Week 3': {'Game1': {'Away': 'Team1', 'Home': 'Team3', 'Venue': 6}}}

To install ndicts pip install ndicts

user_120407510
Related