Question

我从API获取以下格式的词典列表：

例如

xlist =[
    { "id":1, "day":2, name:"abc", ... },
    { "id":1, "day":3, name:"abc", ... },
    { "id":1, "day":2, name:"xyz", ... },
    { "id":1, "day":3, name:"xyz", ... },...
  ]

因此，要将查询存储/优化到DB中，我必须以这种格式转换它们。

生成以下结构的有效方法或其他方法是什么？

unique_xlist =[
  { "id":1, "day":2, name:["abc", "xyz"], ... },
  { "id":1, "day":3, name:["abc", "xyz"], ... },
]

我在做什么：

 names = list(set([ v['name'] for v in xlist])) #-- GET UNIQUE NAMES
 templist =  [{k:(names if k == 'name' else v) \
           for k, v in obj.items()} for obj in xlist] #-- Append Unique names
 unique_xlist= {v['day']:v for v in templist}.values() #- find unique dicts

我不认为这是非常有效的，我使用3个循环只是为了在白天找到唯一的dicts。

Answer 1

您可以使用itertools.groupby：

from itertools import groupby

xlist.sort(key=lambda x: (x["id"], x["day"], x["name"]))  # or use sorted()
unique_xlist = []

for k, g in groupby(xlist, lambda x: (x["id"], x["day"])):
    unique_xlist.append({"id": k[0], "day": k[1], "name": [i["name"] for i in g]})

Answer 2

只需使用使项目唯一的值作为字典的键：

grouped = {}
for x in xlist:
    key = (x['id'], x['day'])
    try:
        grouped[key]['name'].append(x['name'])
    except KeyError:
        grouped[key] = x
        grouped[key]['name'] = [x['name']]

如果有必要，你可以在事后再次解决这个问题。

为词典列表中的两个键创建唯一的词典

2 个答案: