检查python列表中是否存在条目并添加元素

Question

我有两个python形式的列表

list1 = [('TGFB1', 'TGFB1', 1), ('TGFB1', 'CRP', 0.4),('BRCA2', 'TP53', 0.3)]
list2 = [('BRCA1', 'TP53', 2), ('TGFB1', 'CRP', 0.4),('BRCA2', 'TP53', 0.3)]

我需要检查list1中是否存在list2中的每个条目。 如果存在，则添加整数部分并存储在新列表中。 如果不存在，则将该条目添加到新创建的列表中。 所以在这里我新创建的list3应该看起来像

list3 = [('TGFB1', 'TGFB1', 1), ('TGFB1', 'CRP', 0.8),('BRCA2', 'TP53', 0.6),('BRCA1', 'TP53', 2) ]

Answer 1

根据我的上述评论，这可能是您想要的：

from copy import copy

d3 = copy(dict(((x, y), z) for x, y, z in list1))
d2 = dict(((x, y), z) for x, y, z in list2)

for key, value in d2.iteritems():
    d3[key] = (d3[key] if key in d3 else 0.0) + value

如果您确实想翻开d3则返回具有相同结构的列表，请使用：

list3 = [key + (value,) for key, value in d3.iteritems()]

Answer 2

这是我经常做的事情的更复杂的版本。 通常，我不在乎顺序，只是把东西扔进字典。 有时我会按键排序，因此初始顺序无关紧要。

怎么样一个OrderedDict（python2.7 +）：

from collections import OrderedDict

list1 = [('TGFB1', 'TGFB1', 1), ('TGFB1', 'CRP', 0.4),('BRCA2', 'TP53',0.3)]
list2 = [('BRCA1', 'TP53',2), ('TGFB1', 'CRP', 0.4),('BRCA2', 'TP53',0.3)]

d=OrderedDict()
for i in list1:
    d[i[0:2]]=i[2]
for i in list2:
    if i[0:2] in d:
        d[i[0:2]]+=i[2]
    else:
        d[i[0:2]]=i[2]
list3=[k+(v,) for k,v in d.items()]
print(list3)