如何在Python中将此列表转换为字典?


问题内容

我有一个这样的清单:

paths = [['test_data', 'new_directory', 'ok.txt'], ['test_data', 'reads_1.fq'], ['test_data', 'test_ref.fa']]

我想像这样将其转换成字典:

{'test_data': ['ok.txt', 'reads_1.fq'], 'test_data/new_directory', ['ok.txt']}

该列表是动态的。这样做的目的是创建一个简单的树结构。我想这样使用itertools来做到这一点:

from itertools import izip
i = iter(a)
b = dict(izip(i, i))

这样的事情可能吗?谢谢


问题答案:

是的,可以使用collections.defaultdict

>>> from collections import defaultdict
>>> dic = defaultdict(list)
>>> lis = [['test_data', 'new_directory', 'ok.txt'], ['test_data', 'reads_1.fq'], 
for item in lis:                                                                                           
    key = "/".join(item[:-1])
    dic[key].append(item[-1])
...     
>>> dic
defaultdict(<type 'list'>,
{'test_data': ['reads_1.fq', 'test_ref.fa'],
 'test_data/new_directory': ['ok.txt']})

使用简单dict

>>> dic = {}
>>> for item in lis:
    key = "/".join(item[:-1])
    dic.setdefault(key, []).append(item[-1])
...     
>>> dic
{'test_data': ['reads_1.fq', 'test_ref.fa'],
 'test_data/new_directory': ['ok.txt']}