python - 从字符串创建列表字典

Question

我想将诸如 'a=b,a=c,a=d,b=e' 的字符串转换为列表字典 {'a': ['b', 'c', 'd' ], 'b': ['e']} 在 Python 2.6 中。

我目前的解决方案是这样的:

def merge(d1, d2):
    for k, v in d2.items():
        if k in d1:
            if type(d1[k]) != type(list()):
                d1[k] = list(d1[k])
            d1[k].append(v)
        else:
            d1[k] = list(v)
    return d1

record = 'a=b,a=c,a=d,b=e'

print reduce(merge, map(dict,[[x.split('=')] for x in record.split(',')]))

我敢肯定这是不必要的复杂。

有更好的解决方案吗？

Answer 1

d = {}
for i in 'a=b,a=c,a=d,b=e'.split(","):
    k,v = i.split("=")
    d.setdefault(k,[]).append(v)
print d

或者，如果您使用的是 python > 2.4，则可以使用 defaultdict

from collections import defaultdict

d = defaultdict(list)
for i in 'a=b,a=c,a=d,b=e'.split(","):
    k,v = i.split("=")
    d[k].append(v)
print d