Question

for item in listOfModels:
    if item[0] in perms:
        perms[item[0]][item[1]] = True
    else:
        perms[item[0]] = {item[1]: True}

我经常使用这样的代码。请告诉我美丽，简短，正确的方法。

（lib，书籍，样本等）

E.G。我有

[
    ['animal', 'rabbit'],
    ['animal', 'cow'],
    ['plant', 'tree'],
    ['animal', 'elephant'],
    ['fruit', 'strawberry'],
    ['fruit', 'apple'],
]

并且需要

{
    'animal': ['rabbit', 'cow', 'elephant'],
    'plant': ['tree'],
    'fruit': ['strawberry', 'apple'],
}

OR

{
'animal': {
    'rabbit': True,
    'cow': True,
    'elephant': True
},
'plant': {
    'tree': True
},
'fruit': {
    'strawberry': True,
    'apple': True
},

Answer 1

两个选项：使用dict.setdefault()或使用collections.defaultdict() object。

使用dict.setdefault()：

for category, name in listOfModels:
    perms.setdefault(category, {})[name] = True

或使用defaultdict：

from collection import defaultdict

perms = defaultdict(dict)
for category, name in listOfModels:
    perms[category][name] = True

dict.setdefault()为您查找密钥，如果缺少，则使用第二个参数设置值。这样你总是得到一个字典（甚至是空字典），然后你可以在其上设置name密钥。

defaultdict采用工厂参数，每次尝试访问的键都丢失时，将调用工厂以生成默认值。因此，访问perms['missing_key']与使用perms.set_default('missing_key', default)具有相同的效果;根据需要生成新值。

这两种方法都可以轻松地生成列表或集合而不是具有True值的字典：

# producing a list
for category, name in listOfModels:
    perms.setdefault(category, []).append(name)

# or a set
for category, name in listOfModels:
    perms.setdefault(category, set()).append(name)

# same with defaultdict
perms = defaultdict(list)
for category, name in listOfModels:
    perms[category].append(name)

perms = defaultdict(set)
for category, name in listOfModels:
    perms[category].add(name)

这里的集合可能是最好的选择，直接相当于字典值设置为True。

Answer 2

使用setdefault方法。

for item in listOfModels:
    perms.setdefault(item[0], {})[item[1]] = True

如果你知道每个项目只有两个元素，你可以进一步缩短到

for item0, item1 in listOfModels:
    perms.setdefault(item0, {})[item1] = True

如果你想对itertools：

发疯

import operator, itertools
first = operator.itemgetter(0)
second = operator.itemgetter(1)
grouped = itertools.groupby(sorted(listOfModels, key=first), key=first)
result = dict((kind, map(second, examples)) for kind, examples in grouped)

Python语法。美丽，正确，短代码

2 个答案: