从列表元素形成字典

Question

嗨，我有如下列表，其中包含来自图像的元数据如下：

['Component 1: Y component: Quantization table 0, Sampling factors 1 horiz/1 vert', 
 'Component 2: Cb component: Quantization table 1, Sampling factors 1 horiz/1 vert', 
 'Component 3: Cr component: Quantization table 1, Sampling factors 1 horiz/1 vert', 
 'Compression Type: Progressive, Huffman', 'Content-Length: 14312', 'Content-Type: image/jpeg’]

我想做一个字典，用下面的格式拆分列表“：”：

{Component 1: {Y component: [Quantization table 0, Sampling factors 1 horiz/1 vert’], 
 Component 2: {Cb component: [Quantization table 1, Sampling factors 1 horiz/1 vert]}, 
 Component 3: {Cr component: [Quantization table 1, Sampling factors 1 horiz/1 vert]}, 
 Compression Type: [Progressive, Huffman],Content-Length: 14312,Content-Type: image/jpeg}

目前，我已经写了一些代码，这是不工作的。

def make_dict(seq):
res = {}
if seq[0] is not '':
    for elt in seq:
        k, v = elt.split(':')
        try:
            res[k].append(v)  
        except KeyError:
            res[k] = [v]

print res

此代码不起作用。我也尝试过其他方法，但我无法获得格式。

Answer 1

您可以使用collections.OrderedDict在字典理解中使用列表理解

>>> li=['Component 1: Y component: Quantization table 0, Sampling factors 1 horiz/1 vert', 'Component 2: Cb component: Quantization table 1, Sampling factors 1 horiz/1 vert', 'Component 3: Cr component: Quantization table 1, Sampling factors 1 horiz/1 vert', 'Compression Type: Progressive, Huffman', 'Content-Length: 14312', 'Content-Type: image/jpeg']
>>> d=OrderedDict((sub[0],{sub[1]:sub[2:]}) if sub[2:] else (sub[0],sub[1]) for sub in [item.split(':') for item in li])
>>> d
OrderedDict([('Component 1', {' Y component': [' Quantization table 0, Sampling factors 1 horiz/1 vert']}), ('Component 2', {' Cb component': [' Quantization table 1, Sampling factors 1 horiz/1 vert']}), ('Component 3', {' Cr component': [' Quantization table 1, Sampling factors 1 horiz/1 vert']}), ('Compression Type', ' Progressive, Huffman'), ('Content-Length', ' 14312'), ('Content-Type', ' image/jpeg')])
>>> 

Answer 2

l = ['Component 1: Y component: Quantization table 0, Sampling factors 1 horiz/1 vert',
     'Component 2: Cb component: Quantization table 1, Sampling factors 1 horiz/1 vert',
     'Component 3: Cr component: Quantization table 1, Sampling factors 1 horiz/1 vert',
     'Compression Type: Progressive, Huffman', 'Content-Length: 14312', 'Content-Type: image/jpeg']

d = {}

for ele in l:
    spl = ele.split(":", 2)
    if len(spl) == 3:
        k1, k2, v = spl
        d[k1] = {k2: v.split(",")}
    else:
        k,v = spl
        d[k] =   v.split() if "," in v  else v

输出：

{'Component 1': {' Y component': [' Quantization table 0',
                                  ' Sampling factors 1 horiz/1 vert']},
 'Component 2': {' Cb component': [' Quantization table 1',
                                   ' Sampling factors 1 horiz/1 vert']},
 'Component 3': {' Cr component': [' Quantization table 1',
                                   ' Sampling factors 1 horiz/1 vert']},
 'Compression Type': [' Progressive', ' Huffman'],
 'Content-Length': ' 14312',
 'Content-Type': ' image/jpeg'}

要删除空格，您可以将其str.strip掉：

d = {}

for ele in l:
    spl = ele.split(":", 2)
    if len(spl) == 3:
        k1, k2, v = spl
        d[k1] = {k2.strip(): list(map(str.strip,v.split(",")))}
    else:
        k,v = spl
        d[k] = list(map(str.strip, v.split())) if "," in v  else v.strip

输出：

{'Component 1': {'Y component': ['Quantization table 0',
                                 'Sampling factors 1 horiz/1 vert']},
 'Component 2': {'Cb component': ['Quantization table 1',
                                  'Sampling factors 1 horiz/1 vert']},
 'Component 3': {'Cr component': ['Quantization table 1',
                                  'Sampling factors 1 horiz/1 vert']},
 'Compression Type': ['Progressive', 'Huffman'],
 'Content-Length': '14312',
 'Content-Type': 'image/jpeg'}

这两者实际上都与您的预期输出相匹配。

Answer 3

如果你想处理任何级别的字典嵌套，你可以使用像下面这样的递归算法。示例：

def makedict(elem):
    if ':' in elem:
        k,v = map(str.strip, elem.split(':',1))
        return {k:makedict(v)}
    elif ',' in elem:
        elems = list(map(str.strip, elem.split(','))) #Simply map(...) for Python 2.x
        return elems
    return elem

如果你想做一本字典，你可以做-

d = {}
for elem in s:
    d.update(makedict(elem))

或者，如果你想要一个字典列表，在列表理解中为列表中的每个元素调用上面的函数，例如-

result = [makedict(elem) for elem in yourlist]

字典的演示-

>>> d = {}
>>> for elem in s:
...     d.update(makedict(elem))
...
>>> d
{'Component 2': {'Cb component': ['Quantization table 1', 'Sampling fac
>>> import pprint
>>> pprint.pprint(d)
{'Component 1': {'Y component': ['Quantization table 0',
                                 'Sampling factors 1 horiz/1 vert']},
 'Component 2': {'Cb component': ['Quantization table 1',
                                  'Sampling factors 1 horiz/1 vert']},
 'Component 3': {'Cr component': ['Quantization table 1',
                                  'Sampling factors 1 horiz/1 vert']},
 'Compression Type': ['Progressive', 'Huffman'],
 'Content-Length': '14312',
 'Content-Type': 'image/jpeg'}