如何使用其中一个属性作为键从类实例列表创建字典。

时间:2017-02-13 20:52:30

标签: python python-3.x dictionary

我有一个函数(createList),它从文本文件中读取并从文件中的每一行创建类实例。然后,此函数返回类实例列表。现在我想要做的是使用其中一个属性作为键从此列表创建一个字典。

def createDict():
    list = createList()
    fileAsDict = {}
    for i in list:
        fileAsDict[i.name] = i

    return fileAsDict

这似乎是一个简单的解决方案,但我注意到文本文件中的多个实例具有相同的“密钥”。我写的代码不处理这个,并且每次找到相同的i.name时都会覆盖密钥的值。我想要的是它将值存储在列表中,所以当我调用键时,它会打印具有该属性的所有类实例。

我找到了一些像

这样的提示
for key, val in l:
d.setdefault(key, []).append(val)

但我不知道如何将其实现到我的代码中。

1 个答案:

答案 0 :(得分:1)

您使用setdefault走在正确的轨道上。您只需将key替换为i.name即可。这是一个显示逻辑的简单示例实现:

>>> # create a dummy class, so we can put some
>>> # instances in a list
>>> class Dummy:
    def __init__(self, name):
        self.name = name


>>> # create a list with Dummy() class instances. Uh oh! some of them have the
>>> # same value for self.i
>>> classes = [Dummy('a'), Dummy('b'), Dummy('b'), Dummy('c'), Dummy('d')]
>>> 
>>> # now we'll create the dictionary to hold the class instances.
>>> classes_dict = {}
>>> 
>>> # Here we are iterating over the list. For every element in the list,
>>> # we add to the dict using setdefault(). This means that if the element
>>> # key is already in the dict, we append it to the key's list. Otherwise,
>>> # we create a key with a new, empty list.
>>> for each_class in classes:
    classes_dict.setdefault(each_class.name, []).append(each_class)


>>> # final result
>>> classes_dict {'a': [<__main__.Dummy object at 0x0000020B79263550>],
 'b': [<__main__.Dummy object at 0x0000020B792BB1D0>,
     <__main__.Dummy object at 0x0000020B792BB320>],
 'c': [<__main__.Dummy object at 0x0000020B792BB358>],
 'd: [<__main__.Dummy object at 0x0000020B792BB390>]}
>>>