Question

我有四个列表如下：

lista = [['l', 'k'],['e', '3'],['c', 'k'],['x', 'i'],['d', 'f']]
listanum = [1,2,3,4,5]
listb = [['a', 'k'],['c', 'm'],['v', 'f']]
listbnum = [1,3,4]

lista和listanum已同步，listb和listbnum。我想创建一个词典，其中键是listanum中的项目，值是lista和listb中的项目，结果将是：

di = {1: [['l','k'],['a', 'k']], 
      2: [['e', '3'],[]], 
      3:[['c','k'],['c', 'm']], 
      4: [['x', 'i'],['v', 'f']], 
      5: [['d', 'f'][]]
}

因此，如果listb中listanum中的数字没有值，则dictionarys值中的第二个列表为空。

我试过这个：

di = {}
for i in xrange(len(lista)):
  pos = listanum[i]
  if pos not in di:
    di[pos] = [[],[]]
  di[pos][0].append(lista[i])  
  if i in listbnum:
    di[pos][1].append(listb[i])

但收到此错误消息：'IndexError: list index out of range'。我不明白为什么它超出了范围???

Answer 1

In [7]: da = dict(zip(listanum, lista))

In [8]: db = dict(zip(listbnum, listb))

In [9]: {k:[da.get(k,[]), db.get(k,[])] for k in set(listanum + listbnum)}
Out[9]: 
{1: [['l', 'k'], ['a', 'k']],
 2: [['e', '3'], []],
 3: [['c', 'k'], ['c', 'm']],
 4: [['x', 'i'], ['v', 'f']],
 5: [['d', 'f'], []]}

Answer 2

压缩项目，并使用collections.defaultdict将值默认为list s：

from itertools import chain
from collections import defaultdict

di = defaultdict(list)

for key, value in chain(zip(listanum, lista), zip(listbnum, listb)):
    di[key].append(value)

我使用chain更容易在两组键值对上进行循环;这适用于Python 2和3.如果这只是Python 2代码，您可以使用+来连接两个列表。

使用pprint输出并转换回常规dict以简化打印：

>>> pprint(dict(di))
{1: [['l', 'k'], ['a', 'k']],
 2: [['e', '3']],
 3: [['c', 'k'], ['c', 'm']],
 4: [['x', 'i'], ['v', 'f']],
 5: [['d', 'f']]}

这不会为第二组创建空列表;如果你有有空列表，你只能构建两个单独的词典然后合并它们：

dicta = dict(zip(listanum, lista))
dictb = dict(zip(listbnum, listb))

di = {k: [dicta.get(k, []), dictb.get(k, [])] for k in dicta.viewkeys() | dictb.viewkeys()}

对于Python 2，对于Python 3使用.keys()而不是.viewkeys()来生成：

>>> pprint(di)
{1: [['l', 'k'], ['a', 'k']],
 2: [['e', '3'], []],
 3: [['c', 'k'], ['c', 'm']],
 4: [['x', 'i'], ['v', 'f']],
 5: [['d', 'f'], []]}

具体来说，对于您的代码，您将i（lista的索引）与pos混为一谈：

  if i in listbnum:
    di[pos][1].append(listb[i])

对于i = 4，i in listbnum为True，但listb[4]不存在。您的代码还尝试追加来自lista和listb的列表，这不会产生正确的输出。

使用单独的循环为listb / listbnum列表改变您的版本以便工作：

di = {}
for i, pos in enumerate(listanum):
    if pos not in di:
        di[pos] = [[],[]]
    di[pos][0][:] = lista[i]

for i, pos in enumerate(listbnum):
    di[pos][1][:] = listb[i]

从两个不同长度的列表中添加到字典中的值

2 个答案: