python--面试题01

|--题目

写一个函数，将两个dict（key是数字，value是string）进行合并，函数返回合并后的dict。
规则如下：如果一个key仅仅存在于其中一个dict中，则直接加入合并后的dict；
如果一个key在两个dict中都存在，那么给定一个choice值，choice可以是任何string，
如果choice是任一个dict中的value，则写入，否则不写入。

#!/usr/bin/python3

def function(lefdict,rightdict,choice):
    samelist = lefdict.keys() & rightdict.keys() #dict.keys()返回的是一个可迭代对象，取两个dict的keys的交集
    diflist = lefdict.keys() ^ rightdict.keys()  #取两个dict的keys的不同集
    newdict = {}
    for key, value in lefdict.items():
        if key in diflist:
            newdict[key] = value
        elif key in samelist:
            if value == choice:
                newdict[key] = value
    for key, value in rightdict.items():
        if key in diflist:
            newdict[key] = value
        elif key in samelist:
            if value == choice:
                newdict[key] = value
    print(samelist, diflist)
    print(newdict)
    return newdict
    
function({1: 'a', 2: 'b', 3: 'c'}, {4: 'f', 2: 'b', 3: 'd'}, 'b')