Question

我正在寻找pythonic方法来转换看起来像这样的元组列表：

 res = [{type: 1, name: 'Nick'}, {type: 2, name: 'Helma'}, ...]

像这样说：

 {1: [{type: 1, name: 'Nick'}, ...], 2: [{type: 2, name: 'Helma'}, ...]}

现在我用这样的代码（based on this question）执行此操作：

 d = defaultdict(list) 
 for v in res:
    d[v["type"]].append(v)

这是按属性构建对象列表的dict的Pythonic方法吗？

Answer 1

我同意评论员的观点，在这里，列表理解将缺乏，嗯，理解。

话虽如此，这是怎么回事：

import itertools

a = [{'type': 1, 'name': 'Nick'}, {'type': 2, 'name': 'Helma'}, {'type': 1, 'name': 'Moshe'}]
by_type = lambda a: a['type']  
>>> dict([(k, list(g)) for (k, g) in itertools.groupby(sorted(a, key=by_type), key=by_type)])
{1: [{'name': 'Nick', 'type': 1}, {'name': 'Moshe', 'type': 1}], ...}

代码首先按'type'排序，然后使用itertools.groupby按完全相同的标准进行分组。

我写完这篇代码后15秒就停止了解这段代码： - ）

Answer 2

你可以用词典理解来做到这一点，这不像评论所暗示的那样难以理解或不可理解（恕我直言）：

# A collection of name and type dictionaries
res = [{'type': 1, 'name': 'Nick'},
       {'type': 2, 'name': 'Helma'},
       {'type': 3, 'name': 'Steve'},
       {'type': 1, 'name': 'Billy'},
       {'type': 3, 'name': 'George'},
       {'type': 4, 'name': 'Sylvie'},
       {'type': 2, 'name': 'Wilfred'},
       {'type': 1, 'name': 'Jim'}]

# Creating a dictionary by type
res_new = {
    item['type']: [each for each in res
                   if each['type'] == item['type']]
    for item in res
}

>>>res_new
{1: [{'name': 'Nick', 'type': 1},
     {'name': 'Billy', 'type': 1},
     {'name': 'Jim', 'type': 1}],
 2: [{'name': 'Helma', 'type': 2},
     {'name': 'Wilfred', 'type': 2}],
 3: [{'name': 'Steve', 'type': 3},
     {'name': 'George', 'type': 3}],
 4: [{'name': 'Sylvie', 'type': 4}]}

除非我遗漏了某些内容，否则这应该会为您提供您正在寻找的结果。

什么是通过属性构建字典列表的Pythonic方法？

2 个答案: