Я хотел бы сгруппировать по году и городу, накапливая даты и точки, так как на выходе я новичок в python и понятия не имею, как это сделать, у вас есть какие-нибудь идеи, пожалуйста? Я извлек списки list_year и list_city, не уверен, что это полезно? Спасибо

list_pts = [
    {'city' : 'Madrid', 'year' : '2017', 'date' : '05/07/2017', 'pts' : 7},
    {'city' : 'Madrid', 'year' : '2017', 'date' : '14/11/2017', 'pts' : 5},
    {'city' : 'Londres', 'year' : '2018', 'date' : '25/02/2018', 'pts' : 5},
    {'city' : 'Paris', 'year' : '2019', 'date' : '17/04/2019', 'pts' : 4},
    {'city' : 'Londres', 'year' : '2019', 'date' : '15/06/2019', 'pts' : 8},    
    {'city' : 'Paris', 'year' : '2019', 'date' : '21/08/2019', 'pts' : 8},
    {'city' : 'Londres', 'year' : '2019', 'date' : '04/12/2019', 'pts' : 2}]

list_year = ['2017', '2018', '2019']
list_city = ['Paris', 'Madrid', 'Londres']

output =
    [{'year' : '2017', 'city' : 'Madrid', 'date' : ['05/07/2017', '14/11/2017'], 'pts' :[5, 7]},
    {'year' : '2018', 'city' : 'Londres', 'date' : ['25/02/2018'], 'pts' :[5]},
    {'year' : '2019', 'city' : 'Londres', 'date' : ['15/06/2019', '04/12/2019'], 'pts' :[8, 2]},
    {'year' : '2019', 'city' : 'Paris', 'date' : ['17/04/2019', '21/08/2019'], 'pts' :[4, 8]}]
2
Maxime Lara 15 Окт 2021 в 18:54

3 ответа

Лучший ответ

Вы можете использовать groupby() для группировки слов в сочетание с itemgetter():

from itertools import groupby
from operator import itemgetter

list_pts = [
    {'city': 'Madrid', 'year': '2017', 'date': '05/07/2017', 'pts': 7},
    {'city': 'Madrid', 'year': '2017', 'date': '14/11/2017', 'pts': 5},
    {'city': 'Londres', 'year': '2018', 'date': '25/02/2018', 'pts': 5},
    {'city': 'Paris', 'year': '2019', 'date': '17/04/2019', 'pts' : 4},
    {'city': 'Londres', 'year': '2019', 'date': '15/06/2019', 'pts': 8},
    {'city': 'Paris', 'year': '2019', 'date': '21/08/2019', 'pts': 8},
    {'city': 'Londres', 'year': '2019', 'date': '04/12/2019', 'pts': 2}
]

city_year_getter = itemgetter('city', 'year')
date_pts_getter = itemgetter('date', 'pts')

result = []
for (city, year), objs in groupby(sorted(list_pts, key=city_year_getter), 
                                  city_year_getter):
    dates, ptss = zip(*map(date_pts_getter, objs))
    result.append({
        'city': city,
        'year': year,
        'date': list(dates),
        'pts': list(ptss)
    })
1
Olvin Roght 15 Окт 2021 в 16:23

Один подход, основанный на использовании collections.defaultdict для группировки значений необходимо сделать следующее:

from collections import defaultdict
from operator import itemgetter
import pprint

list_pts = [
    {'city': 'Madrid', 'year': '2017', 'date': '05/07/2017', 'pts': 7},
    {'city': 'Madrid', 'year': '2017', 'date': '14/11/2017', 'pts': 5},
    {'city': 'Londres', 'year': '2018', 'date': '25/02/2018', 'pts': 5},
    {'city': 'Paris', 'year': '2019', 'date': '17/04/2019', 'pts': 4},
    {'city': 'Londres', 'year': '2019', 'date': '15/06/2019', 'pts': 8},
    {'city': 'Paris', 'year': '2019', 'date': '21/08/2019', 'pts': 8},
    {'city': 'Londres', 'year': '2019', 'date': '04/12/2019', 'pts': 2}]

# function for extracting city and year (to be used as a grouping key)
city_and_year = itemgetter("city", "year")

# function for extracting dates and points 
date_and_points = itemgetter("date", "pts")

# group by key (city, year) by using a defaultdict
res = defaultdict(list)
for record in list_pts:
    res[city_and_year(record)].append(date_and_points(record))

# transform to the desired format
result = []
for (city, year), values in res.items():
    dates, points = zip(*values)
    result.append({"city": city, "year": year, "dates": list(dates), "pts": list(points)})

# use pprint to nicely print the output
pprint.pprint(result)

Вывод

[{'city': 'Madrid',
  'dates': ['05/07/2017', '14/11/2017'],
  'pts': [7, 5],
  'year': '2017'},
 {'city': 'Londres', 'dates': ['25/02/2018'], 'pts': [5], 'year': '2018'},
 {'city': 'Paris',
  'dates': ['17/04/2019', '21/08/2019'],
  'pts': [4, 8],
  'year': '2019'},
 {'city': 'Londres',
  'dates': ['15/06/2019', '04/12/2019'],
  'pts': [8, 2],
  'year': '2019'}]
1
Dani Mesejo 15 Окт 2021 в 16:15
list_pts = [
  {'city': 'Madrid',  'year' : '2017', 'date' : '05/07/2017', 'pts' : 7},
  {'city': 'Madrid',  'year' : '2017', 'date' : '14/11/2017', 'pts' : 5},
  {'city': 'Londres', 'year' : '2018', 'date' : '25/02/2018', 'pts' : 5},
  {'city': 'Paris',   'year' : '2019', 'date' : '17/04/2019', 'pts' : 4},
  {'city': 'Londres', 'year' : '2019', 'date' : '15/06/2019', 'pts' : 8},
  {'city': 'Paris',   'year' : '2019', 'date' : '21/08/2019', 'pts' : 8},
  {'city': 'Londres', 'year' : '2019', 'date' : '04/12/2019', 'pts' : 2}]

group_by = {}
for pt in list_pts:
  d = group_by.setdefault((pt['year'], pt['city']), {})
  for k, v in pt.items():
    if k not in {'year', 'city'}:
      d.setdefault(k, []).append(v)

output = []
for (year, city), rest in group_by.items():
  output.append({'year': year, 'city': city} | rest)

Для Python до 3.9 см. этот ответ о слиянии словарей.

Это вопрос использования пары (year, city) в качестве ключа (которая может быть одной с кортежами в Python) и объединения всех остальных элементов в списки. Превращение из group_by в output - всего лишь косметическая вещь.

0
Andrej Podzimek 15 Окт 2021 в 16:21