按唯一值分组列表

2024-03-29 10:41:35 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一张单子

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'}]

我该怎么做? 有什么想法吗?你知道吗


Tags: albumartistlistmomo单子pulsedamntimes
3条回答
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'}]
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]  
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

相关问题 更多 >