在 python - 我如何将项目转换为字典？

Question

我有以下项目列表（键值对）：

items = [('A', 1), ('B', 1), ('B', 2), ('C', 3)]

我想得到什么：

{
  'A' : 1,
  'B' : [1,2]
  'C' : 3
}

我天真的解决方案：

res = {}
for (k,v) in items:
     if k in res:
         res[k].append(v)
     else:
         res[k] = [v]

我正在寻找一些优化的更 Pythonic 的解决方案，有人吗？

Answer 1

使用这里可以使用defaultdict。

from collections import defaultdict

res = defaultdict(list)
for (k,v) in items:
    res[k].append(v)
# Use as dict(res)

编辑：

这是使用groupby的方法，但请注意， 上面的方法更干净，更整洁 ：

>>> data = [('A', 1), ('B', 1), ('B', 2), ('C', 3)]
>>> dict([(key,list(v[1] for v in group)) for (key,group) in groupby(data, lambda x: x[0])])
{'A': [1], 'C': [3], 'B': [1, 2]}

缺点 ：每个元素都是一个列表。 根据需要将列表更改为生成器。

---

要将所有单个项目列表转换为单个项目：

>>> res = # Array of tuples, not dict
>>> res = [(key,(value[0] if len(value) == 1 else value)) for key,value in res]
>>> res
[('A', 1), ('B', [1, 2]), ('C', 3)]

Answer 2

如果您不想使用defaultdict / groupby ，则可以使用以下方法：

d = {}
for k,v in items:
   d.setdefault(k, []).append(v)

Answer 3

使用defaultdict非常容易做到这一点，可以从collections导入defaultdict 。

>>> from collections import defaultdict
>>> items = [('A', 1), ('B', 1), ('B', 2), ('C', 3)]
>>> d = defaultdict(list)
>>> for k, v in items:
    d[k].append(v)
>>> d
defaultdict(<class 'list'>, {'A': [1], 'C': [3], 'B': [1, 2]})

您还可以使用字典理解：

>>> d = {l: [var for key, var in items if key == l] for l in {v[0] for v in items}}
>>> d
{'A': [1], 'C': [3], 'B': [1, 2]}

Answer 4

您可以使用WebOb multidict实现：

 >>> from webob import multidict
 >>> a = multidict.MultiDict(items)
 >>> a.getall('B')
 [1,2]

Answer 5

也许看起来很丑，但是行得通。

In [1]: items = [('A', 1), ('B', 1), ('B', 2), ('C', 3)]

In [2]: d = {}

In [3]: map(lambda i: d.update({i[0]: i[1] if d.get(i[0], i[1]) == i[1] else [d[i[0]], i[1]]}), items)
Out[3]: [None, None, None, None]

In [4]: print d
{'A': 1, 'C': 3, 'B': [1, 2]}

在else分支中，我们可以检查d[i[0]]返回列表。

Answer 6

将字典转换为项目列表

dict_items = list(dict_1.items())

将项目列表转换回字典

dict2 = dict(dict_items)