合并功能仅适用于有序列表

时间:2017-09-14 11:43:55

标签: python python-3.x list

我将这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 

^^那可行,还在测试!

1 个答案:

答案 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]]