将列表列表分解为词典

时间:2015-01-19 23:31:05

标签: python python-2.7 dictionary-comprehension

我有一个列表列表s,这是在Fruit上查询数据库的结果,item[0]是水果的名称,item[1]是否是水果有种子,item[2]是否可以食用。

s = [['Apple','Yes','Edible'], ['Watermellon','Yes','Yes']]

由于我的实际列表要大得多,我想要一种非常简单的方法来引用/返回这些值。例如,print my_dict['Apple']['Seeds']会产生Yes

我认为我最好的选择是创建一个字典,但我正在寻找关于这是一个好方法以及如何做的建议。

我开始编写一些代码,但我不确定如何获得第二组标题,所以我的示例使用了索引。

my_dict =  {t[0]:t[1:] for t in s}

print my_dict['Apple'][0]

3 个答案:

答案 0 :(得分:5)

fruit_map = {
    fruit: {'Seeds': seeds, 'Edible': edible} for fruit, seeds, edible in s}

答案 1 :(得分:2)

如果第二组键永远不会改变,最好用字段定义一个合适的对象。这可能看起来有些过分或冗长,但总有collections.namedtuple可以提供帮助。

namedtuple从字段名称列表中创建一个新类。该类还支持由列表初始化。要使用您的示例:

import collections

Fruit = collections.namedtuple('Fruit', ['name', 'seeds', 'edible'])

这样,您可以轻松地从列表中创建Fruit个对象:

f = Fruit('Apple', True, False)
# Or, if you already have a list with the values
params = ['Apple', True, False]
f = Fruit(*params)

print f.seed

所以你可以用一种非常简单的方式创建一个水果列表:

s = [['Apple','Yes','Edible'], ['Watermellon','Yes','Yes']]

fruits = [Fruit(*l) for l in s]

你真的需要一个由某个字段索引的字典,它没有太大区别:

s = [['Apple','Yes','Edible'], ['Watermellon','Yes','Yes']]

fruit_dict = {l[0]: Fruit(*l) for l in s}    
print(fruit_dict['Apple'].seeds)
将值列表转换为更易于使用的对象时(例如,在读取CSV文件时,这种情况非常类似于您所要求的情况),

namedtuple可以非常方便。

答案 2 :(得分:0)

import copy

def list_to_dict(lst):
    local = copy.copy(lst) # copied lst to local
    fruit = [i.pop(0) for i in local] # get fruit names

    result = {}
    for i in range(len(local)):
        result[fruit[i]] = local[i]
    return result

这将返回您想要的字典。