List comprehension - group data based on occurance of specific elements in a list

Question

I have a list like below :

[' Premier League 
', ' Fulham ', ' 2 ', ' Liverpool ', ' 2 ', ' Bournemouth ', ' 2 ', ' Aston Villa ', ' 0 ', ' Leeds ', ' 2 ', ' Wolves ', ' 1 ', ' Newcastle United ', ' 2 ', ' Nottingham Forest ', ' 0 ', ' Tottenham ', ' 4 ', ' Southampton ', ' 1 ', ' Everton ', ' 0 ', ' Chelsea ', ' 1 ', ' Bundesliga 
', ' 1. FC Union Berlin ', ' 3 ', ' Hertha BSC ', ' 1 ', ' M\'gladbach ', ' 3 ', ' Hoffenheim ', ' 1 ', ' Augsburg ', ' 0 ', ' SC Freiburg ', ' 4 ', ' VfL Bochum ', ' 1 ', ' Mainz 05 ', ' 2 ', ' VfL Wolfsburg ', ' 2 ', ' Werder Bremen ', ' 2 ', ' Borussia Dortmund ', ' 1 ', ' Bayer Leverkusen ', ' 0 ', ' Scottish Premiership 
', ' Aberdeen ', ' 4 ', ' St. Mirren ', ' 1 ', ' Motherwell ', ' 1 ', ' St. Johnstone ', ' 2 ', ' Rangers ', ' 2 ', ' Kilmarnock ', ' 0 ', ' Ross County ', ' 1 ', ' Celtic ', ' 3 ', ' Ligue 1 Uber Eats 
', ' Strasbourg ', ' 1 ', ' Monaco ', ' 2 ', ' Clermont ', ' 0 ', ' PSG ', ' 5 ']

I am trying to extract data of a few top leagues and want to discard others. Following another example I have this code :

leagues = (['Premier League', 'Spanish La Liga', 'Bundesliga', 'Italian Serie A','Ligue 1 Uber Eats', 'Champions League'])

data = [[l[l.index(left) + len(left):l.index(right)] for l in data if i in l] for i in leagues]

But I am not getting the expected result like which should be like below :

[['Premier League', * all matches of PL], ['Bundesliga', * all Bundesliga matches]].

Please help me with this as I have been burning my head over this for quite a long time now.

Thanks

Nick · Accepted Answer

You can iterate the values in your list, updating league, teams and scores when you see the matching tag, and then writing a match to the result when you have two teams and two scores. I've created a dict of matches with the league as the key, it should be reasonably easy to change the format if you want something else (e.g. with list(result.items()))

import re
from collections import defaultdict

result = defaultdict(list)
for d in data:
    l = re.search(r'>([^<]+)', d)
    if l is not None:
        league = l.group(1).strip()
        teams = []
        scores = []
        continue
    t = re.search(r'name">([^<]+)', d)
    if t is not None:
        teams.append(t.group(1).strip())
        continue
    s = re.search(r'score">\s*(\d+)\s*', d)
    if s is not None:
        scores.append(int(s.group(1)))
        if len(scores) == 2:
            result[league].append({ 'teams' : teams[:], 'scores' : scores[:] })
            teams = []
            scores = []

Output (for your sample data):

{
  'Premier League': [
    {'teams': ['Fulham', 'Liverpool'], 'scores': [2, 2]},
    {'teams': ['Bournemouth', 'Aston Villa'], 'scores': [2, 0]},
    {'teams': ['Leeds', 'Wolves'], 'scores': [2, 1]},
    {'teams': ['Newcastle United', 'Nottingham Forest'], 'scores': [2, 0]},
    {'teams': ['Tottenham', 'Southampton'], 'scores': [4, 1]},
    {'teams': ['Everton', 'Chelsea'], 'scores': [0, 1]}
  ],
  'Bundesliga': [
    {'teams': ['1. FC Union Berlin', 'Hertha BSC'], 'scores': [3, 1]}, 
    {'teams': ["M'gladbach", 'Hoffenheim'], 'scores': [3, 1]},
    {'teams': ['Augsburg', 'SC Freiburg'], 'scores': [0, 4]},
    {'teams': ['VfL Bochum', 'Mainz 05'], 'scores': [1, 2]},
    {'teams': ['VfL Wolfsburg', 'Werder Bremen'], 'scores': [2, 2]},
    {'teams': ['Borussia Dortmund', 'Bayer Leverkusen'], 'scores': [1, 0]}
  ],
  'Scottish Premiership': [
    {'teams': ['Aberdeen', 'St. Mirren'], 'scores': [4, 1]},
    {'teams': ['Motherwell', 'St. Johnstone'], 'scores': [1, 2]}
    {'teams': ['Rangers', 'Kilmarnock'], 'scores': [2, 0]},
    {'teams': ['Ross County', 'Celtic'], 'scores': [1, 3]}
  ],
  'Ligue 1 Uber Eats': [
    {'teams': ['Strasbourg', 'Monaco'], 'scores': [1, 2]},
    {'teams': ['Clermont', 'PSG'], 'scores': [0, 5]}
  ]
}

List comprehension - group data based on occurance of specific elements in a list

Answers (1)

Related Questions