我将这2个列表作为输入:
list1 = [['A', 14, 'I', 10, 20], ['B', 15, 'S', 30, 40], ['C', 16, 'F', 50, 60]]
list2 = [['A', 14, 'Y', 0, 200], ['B', 15, 'M', 0, 400], ['C', 17, 'G', 0, 600]]
我想要的输出是这样的:
finalList = [['A', 14, 'Y', 10, 200], ['B', 15, 'M', 30, 400], ['C', 16, 'F', 50, 60],['C', 17, 'G', 0, 600]]
使用此功能:
def custom_merge(list1, list2):
finalList = []
for sub1, sub2 in zip(list1, list2):
if sub1[1]==sub2[1]:
out = sub1.copy()
out[2] = sub2[2]
out[4] = sub2[4]
finalList.append(out)
else:
finalList.append(sub1)
finalList.append(sub2)
return finalList
我确实会得到我想要的输出,但如果我切换位置(list2 [1]和list2 [2])和我的list2:
list2 = [['A', 14, 'Y', 0, 200], ['C', 17, 'G', 0, 600], ['B', 15, 'M', 0, 400]]
然后输出将是:
[['A', 14, 'Y', 10, 200], ['B', 15, 'S', 30, 40], ['C', 17, 'G', 0, 600], ['C', 16, 'F', 50, 60], ['B', 15, 'M', 0, 400]]
(注意额外的['B',15,'M',0,400])
如果我的列表在列表列表中有不同的顺序,我必须在我的函数中修改以获得我的第一个所需输出!我使用python 3.谢谢!
LATER EDIT:
合并规则:
当list1[listindex][1] == list2[listindex][1]
(例如:14==14
时),请在list1
- >中替换list2[2]
和list2[4]
(例如:'Y'和200),如果不是只是将list2
中不匹配的列表添加到list1
(就像在我想要的输出中一样)并保留list1中不匹配的那些(例如:['C', 16, 'F', 50, 60]
)
需要注意的是list1和list2可以有不同的len(list1可以有比list2更多的列表,反之亦然)
EDIT.2 我发现了这个:
def combine(list1,list2):
combined_list = list1 + list2
final_dict = {tuple(i[:2]):tuple(i[2:]) for i in combined_list}
merged_list = [list(k) + list (final_dict[k]) for k in final_dict]
return merged_list
^^那可行,还在测试!
答案 0 :(得分:1)
您可以在合并列表之前按照子列表中的第一个元素对列表进行排序。
def custom_merge(list1, list2):
finalList = []
for sub1, sub2 in zip(sorted(list1), sorted(list2)):
if sub1[1]==sub2[1]:
out = sub1.copy()
out[2] = sub2[2]
out[4] = sub2[4]
finalList.append(out)
else:
finalList.append(sub1)
finalList.append(sub2)
return finalList
测试:
list1 = [['A', 14, 'I', 10, 20], ['B', 15, 'S', 30, 40], ['C', 16, 'F', 50, 60]]
list2 = [['A', 14, 'Y', 0, 200], ['C', 17, 'G', 0, 600], ['B', 15, 'M', 0, 400]]
custom_merge(list1, list2)
# returns:
[['A', 14, 'Y', 10, 200],
['B', 15, 'M', 30, 400],
['C', 16, 'F', 50, 60],
['C', 17, 'G', 0, 600]]