Python字典:根据其他值中的键出现替换值

Question

如果我没有弄错的话,我的问题可能包括理解.
所以我有一本字典:

data = {
 1: [2, 4], 
 2: [3], 
 3: [1, 2, 6], 
 4: [1, 3, 6], 
 6: [3, 4]
}  

我希望重新排列字典值(而不是键),以便在任何其他值中每次出现"键"时,当前键的值现在应该包含最初发生的值的"键".
例如,字典现在应该如下所示:

new_data = {
 1: [3, 4], 
 2: [1, 3], 
 3: [2, 4, 6], 
 4: [1, 6], 
 6: [3, 4]
}  

我确定我必须使用另一个空字典来解决这个问题,我必须使用循环.我有一个起点,但效率不高:

for pk,pv in zip(data.keys(), data.values()):
    #print(pk,pv)
    for pvs in pv:
        #print(pvs)
        if pk == pvs:
            new_data.setdefault(pk, [])
            new_data[pk].append(pvs)

print(new_data)  

任何帮助表示赞赏.在此先感谢:)
(使用Ubuntu 14.04 32位VM和Python 2.7)

Answer 1

也许这不是最有效的方法,但我认为这是最好的阅读之一:).

我之前应该说的一件事是,如果可以将初始字典值改为set而不是list,那么可能会导致一些性能提升,因为set成员资格检查可能比listss 更快.

开始:

def present_in(k, d):
    return {idx for idx in d if k in d[idx]}

new_data = {k: present_in(k, data) for k in data}

基本上,该present_in函数创建一组与键相对应的值,其中键k作为成员存在.然后我们使用它来new_data使用字典理解来创建字典.请再次注意,此字典的值是一个集合,而不是列表.list()如果需要列表,可以轻松地将它们包装起来.

编辑:

如果您希望将new_data键可能作为键丢失data但作为其中一个值存在,则可以更改代码以首先创建一组所有可能的键:

all_keys = {v for vs in data.values() for v in vs}

def present_in(k, d):
    return {idx for idx in d if k in d[idx]}

new_data = {k: present_in(k, data) for k in all_keys}