Python翻译字典中的列表/字典

Question

Sorry, I didn't really know what to put in title, I hope the example will be enough

Case 1 - list

I have an object this format:

{
    'key[0]': 'value 0',
    'key[1]': 'value 1',
    'key[2]': 'value 2'
}

And I want to get an object like this:

{
    'key': [
        'value 0',
        'value 1',
        'value 2'
    ]
}

Case 2 - dict

Another object I could receive is something like this:

{
    'key[subkey0]': 'value 0',
    'key[subkey1]': 'value 1',
    'key[subkey2]': 'value 2'
}

And I want to get:

{
    'key': {
        'subkey0': 'value 0',
        'subkey1': 'value 1',
        'subkey2': 'value 2'
    }
}

---

My question is, is there any lib that does that? If not, what is the most effective/optimized way to do that?

You can consider that in the first case the list I received will be always complete, that means that no number will be skipped

Answer 1

Here's a start:

from collections import defaultdict
import re
import pprint

def convert(d):
    result = defaultdict(dict)
    for key, value in d.items():
        main, sub = re.match(r'(\w+)\[(\w+)\]', key).groups()
        result[main][sub] = value

    result = dict(result)  # just for display really
    pprint.pprint(result)

convert({
    'key1[subkey0]': 'value 0',
    'key1[subkey1]': 'value 1',
    'key2[subkey0]': 'value 2',
    'key2[subkey1]': 'value 3',
})

Output:

{'key1': {'subkey0': 'value 0', 'subkey1': 'value 1'},
 'key2': {'subkey0': 'value 2', 'subkey1': 'value 3'}}

Answer 2

For the first case,

old = {
    'key[0]': 'value 0',
    'key[1]': 'value 1',
    'key[2]': 'value 2'
}
new = dict()
the_key = old.keys()[0].split("[")[0]
new[the_key] = list()
for key in old:
    new[the_key].append(old[key])

print new

Splitted one of the old dictionary keys by [ to get the real key. Then constructed the new dict using this key. It strikes me as a bad practice, yet it works. So you can take a look.