如何将两组列表组合到字典中

时间:2013-03-15 21:41:25

标签: python dictionary

我有两组具有相似数据的列表,我需要将它们组合成一个字典。该列表必须包含两组信息,填写在另一个具有" - "的任何地方。到目前为止,这就是我所拥有的......

aList = [['Last Name', 'First Name', '2000', '2012'], 
       ['Roberts', 'Gloria', '-', '123000'], 
       ['Arreguin', 'Jeffrey', '81000', '-'], 
       ['Myers', 'George', '-', '86000'], 
       ['Willis', 'Mabel', '112000', '-'], 
       ['Oneal', 'Kevin', '96000', '77000'], 
       ['Paz', 'Barbara', '-', '77000'], 
       ['Franklin', 'Claude', '94000', '-'], 
       ['Bradley', 'Shannon', '89000', '-'], 
       ['Fix', 'Anna', '76000', '126000'], 
       ['Meyer', 'Loretta', '-', '116000'], 
       ['Daniels', 'Christina', '85000', '-'], 
       ['Graham', 'Veronica', '-', '136000']]

newList = [['Last Name', 'First Name', '2000', '2012'], 
       ['Meyer', 'Loretta', '123000', '-'], 
       ['Arreguin', 'Jeffrey', '81000', '-'], 
       ['Mielke', 'George', '137000', '-'], 
       ['Thomas', 'Lewis', '132000', '-'], 
       ['Harper', 'Crystal', '80000', '-'], 
       ['Young', 'Gary', '-', '94000'], 
       ['Franklin', 'Claude', '94000', '-'], 
       ['Hedrick', 'James', '-', '105000'], 
       ['Bradley', 'Shannon', '89000', '-'], 
       ['Thigpen', 'Michael', '79000', '-'], 
       ['Willis', 'Mabel', '112000', '-'], 
       ['Hullinger', 'Molly', '70000', '-'], 
       ['Myers', 'George', '-', '86000'], 
       ['Paz', 'Barbara', '-', '77000'], 
       ['Edwards', 'Kathryn', '117000', '97000'], 
       ['Roberts', 'Gloria', '-', '123000'], 
       ['Daniels', 'Christina', '-', '137000'], 
       ['Graham', 'Veronica', '-', '136000']]

def mergeData(newList,aList):
    aDict={}
    for item in range(1,len(newList)):
        key=(newList[item][0],newList[item][1])
        value=(newList[item][2],newList[item][3])
        aDict[key]=value

    for item in range(1,len(aList)):
        #stuck here not sure if im going right way with this 


print(aDict)

这是一个样本

newList= ['Meyer', 'Loretta', '123000', '-']
aList=['Meyer', 'Loretta', '-', '116000']

所以组合字典应该是

{(‘Meyer’, ‘Loretta’): (‘123000’,’116000’)}

此条目。

1 个答案:

答案 0 :(得分:1)

只需遍历第一个列表即可构建一个带有字典理解的字典:

aDict = {(e[0], e[1]): (e[2], e[3]) for e in aList[1:]}

下一个循环遍历另一个,使用辅助函数根据-值选择一个或另一个:

def pick(vals):
    return vals[1] if vals[0] == '-' else vals[0]

for e in newList:
    key = (e[0], e[1])
    existing = aDict.get(key, ('-', '-'))
    value = (e[2], e[3])
    aDict[key] = tuple(map(pick, zip(existing, value)))

您的第一个输入结果为:

{('Arreguin', 'Jeffrey'): ('81000', '-'),
 ('Bradley', 'Shannon'): ('89000', '-'),
 ('Daniels', 'Christina'): ('85000', '137000'),
 ('Edwards', 'Kathryn'): ('117000', '97000'),
 ('Fix', 'Anna'): ('76000', '126000'),
 ('Franklin', 'Claude'): ('94000', '-'),
 ('Graham', 'Veronica'): ('-', '136000'),
 ('Harper', 'Crystal'): ('80000', '-'),
 ('Hedrick', 'James'): ('-', '105000'),
 ('Hullinger', 'Molly'): ('70000', '-'),
 ('Last Name', 'First Name'): ('2000', '2012'),
 ('Meyer', 'Loretta'): ('123000', '116000'),
 ('Mielke', 'George'): ('137000', '-'),
 ('Myers', 'George'): ('-', '86000'),
 ('Oneal', 'Kevin'): ('96000', '77000'),
 ('Paz', 'Barbara'): ('-', '77000'),
 ('Roberts', 'Gloria'): ('-', '123000'),
 ('Thigpen', 'Michael'): ('79000', '-'),
 ('Thomas', 'Lewis'): ('132000', '-'),
 ('Willis', 'Mabel'): ('112000', '-'),
 ('Young', 'Gary'): ('-', '94000')}

所有在你的合并函数中再次组合在一起:

def pick(vals):
    return vals[1] if vals[0] == '-' else vals[0]

def mergeData(newList, aList):
    aDict = {(e[0], e[1]): (e[2], e[3]) for e in aList[1:]}

    for e in newList:
        key = (e[0], e[1])
        existing = aDict.get(key, ('-', '-'))
        value = (e[2], e[3])
        aDict[key] = tuple(map(pick, zip(existing, value)))

    return aDict