[英]Grouping list by unique values
我有一個清單
list = [{'album': 'Spring Times', 'artist': 'Momo Pulse'}, {'album': 'Spring Times', 'artist': 'K.oshkin'}, {'album': 'Damn ', 'artist': 'Florent B'}]
我想將其分組以獲得:
list = [{'album': 'Spring Times', 'artist1': 'Momo Pulse', 'artist2': 'K.oshkin'}, {'album': 'Damn ', 'artist1': 'Florent B'}]
我怎樣才能做到這一點? 有任何想法嗎?
from itertools import groupby
# input
my_list = [{'album': 'Spring Times', 'artist': 'Momo Pulse'}, {'album': 'Spring Times', 'artist': 'K.oshkin'}, {'album': 'Damn ', 'artist': 'Florent B'}]
# Have a function to return the merged dictionary after an update
def merge_dict(a, b):
a.update(b)
return a
# key function for sort and groupby
sortkey = lambda d: d['album']
# Sort and group by album
my_groups = groupby(sorted(my_list, key=sortkey), key=sortkey)
# Generate output
print [merge_dict({'album':k},{'artist'+str(i+1):d['artist'] for i, d in enumerate(g)}) for k, g in my_groups]
from collections import defaultdict
l = [{'album': 'Spring Times', 'artist': 'Momo Pulse'},
{'album': 'Spring Times', 'artist': 'K.oshkin'},
{'album': 'Damn ', 'artist': 'Florent B'}]
d = defaultdict(list)
for record in l:
d[record['album']].append(record['artist'])
現在,我們有了一個字典,將專輯名稱映射到藝術家列表。
final = []
for album, artists in d.items():
temp = {'album': album}
for i, x in enumerate(artists, start=1):
temp['artist{}'.format(i)] = x
final.append(temp)
print(final)
版畫
[{'album': 'Damn ', 'artist1': 'Florent B'}, {'album': 'Spring Times', 'artist1': 'Momo Pulse', 'artist2': 'K.oshkin'}]
l = [{'album': 'Spring Times', 'artist': 'Momo Pulse'}, {'album': 'Spring Times', 'artist': 'K.oshkin'}, {'album': 'Damn ', 'artist': 'Florent B'}]
albums = {}
for el in l:
album = albums.setdefault(el['album'], {})
artist_id = sum(1 for k in album if k.startswith('artist'))
album['artist'+str(artist_id)] = el['artist']
l2 = albums.values() #This is your output
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.