我有一个列表列表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]
答案 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
这将返回您想要的字典。