从字符串中创建一个列表字典

Question

我希望将类似'a=b,a=c,a=d,b=e'的字符串转换为Python 2.6中的字典列表{'a': ['b', 'c', 'd'], 'b': ['e']}。涉及到IT技术相关内容。以下是我的当前解决方案：

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(',')]))

我相信这是不必要的复杂。有更好的解决方案吗？

- Chris Card

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版本的Python，您可以使用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

- Shawn Chin

Answer 2

>>> result={}
>>> mystr='a=b,a=c,a=d,b=e'
>>> for k, v in [s.split('=') for s in mystr.split(',')]:
...     result[k] = result.get(k, []) + [v]
... 
>>> result
{'a': ['b', 'c', 'd'], 'b': ['e']}

- kojiro

Answer 3

怎么样...

STR = "a=c,b=d,a=x,a=b"
d = {} # An empty dictionary to start with.

# We split the string at the commas first, and each substr at the '=' sign
pairs = (subs.split('=') for subs in STR.split(','))

# Now we add each pair to a dictionary of lists.
for key, value in pairs:
    d[key] = d.get(key, []) + [value]

- salezica

Answer 4

使用正则表达式可以在仅一次操作中完成两次拆分的工作：

import re

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

dic = {}
for k,v in re.findall('(\w+)=(\w+)\s*(?:,|\Z)',ch):
    dic.setdefault(k,[]).append(v)

print dic

- eyquem