我有两个清单：
list1 = [ {'sth': 13,'important_key1': 'AA','important_key2': '3'},{'oh!': 14,'important_key1': 'FF','important_key2': '4'},{'sth_else': 'abc','important_key1': 'ZZ','important_key2': '5'}]
list2 = [ {'why-not': 'tAk','important_key1': 'GG',{'hmmm': 'no','important_key2': '3'}]
我想返回一个仅包含来自list1的对象的列表,但如果同一个important_key1和important_key2在list2中的任何元素中,我想从list2中获取该元素.
所以输出应该是：
[ {'hmmm': 'no','important_key2': '5'}]
通过两个或三个循环来做它并不复杂,但我想知道是否有一种简单的方法可以使用列表推导或类似的东西.
这是“正常”的方式：
list1 = [ {'sth': 13,'important_key2': '4'}]
list2 = [ {'hmmm': 'no',{'why-not': 'tAk','important_key2': '4'}]
final_list = []
for element in list1:
there_was_in_list2 = False
for another_element in list2:
if element['important_key1'] == another_element['important_key1'] and element['important_key2'] == another_element['important_key2']:
final_list.append(another_element)
there_was_in_list2 = True
break
if not there_was_in_list2:
final_list.append(element)
print(final_list)
是否有任何Pythonic方法可以做到这一点？
解决方法
您可以将list2转换为由list2中的重要键的值元组索引的dict,然后使用它来确定list1中的相同键是否具有与列表推导中的list1迭代相同的值,以便时间复杂性从你的O(n * m)减少到O(n)：
keys = ['important_key1','important_key2']
d2 = {tuple(d[k] for k in keys): d for d in list2[::-1]}
print([d2.get(tuple(d[k] for k in keys),d) for d in list1])
这输出(带有您的样本输入)：
[{‘hmmm’：’不’,’important_key1’：’AA’,’important_key2’：’3′},{‘哦！’：14,’important_key1’：’FF’,’important_key2’：’4′ },{‘sth_else’：’abc’,’important_key1’：’ZZ’,’important_key2’：’5′}]
正如您在问题中所描述的那样,list1中只有{‘sth’：13,’important_key2’：’3′}将替换为{‘hmmm’：’no’,’important_key1’： ‘AA’,’important_key2’：’3′}因为只有这个dict的important_key1和important_key2都匹配list2中的dict.