python：列表求差

逃离我推掉我的手 2022-11-29 12:15 158阅读 0赞

列表格式：

result1 = [("2020-01",21,12,12,23,34),("2020-02", 23,34,45,56,67),("2020-05",1,2,3,4,4)]
    result2 = [("2020-01",43,45,56,23,34),("2020-02", 23,12,23,34,45),("2020-05",12,21,31,41,14),("2020-06",11,21,31,14,14)]
    
    or
    
    result1 = [("2020-01",21,12,12,23,34),("2020-02", 23,34,45,56,67),("2020-05",1,2,3,4,4)]
    result2 = [("2020-01",43,45,56,23,34),("2020-02", 23,12,23,34,45),("2020-05",12,21,31,41,14)]

需要求result2 和result1列表各值之差。  
背景：result1的长度小于等于result2，且result1中月份一定会在result2中出现，但result2中的月份不一定存在于result1中。

一开始的思路是判断二者长度是否相同，如果相同，代表二者出现的年月相同，不同则代表result2的值多了一些月份。

所以写了两个方法，分别判断二者长度相同和不同的情况。

def get_result_with_same_length(result1, result2):
    """二者长度相同时"""
        result_list = []
        result_all = zip(result1, result2)
        for result_1, result_2 in result_all:
            i = 0
            list_one = []
            list_one.append(result_1[i])
            while i < len(result_1) - 1:
                i += 1
                list_one.append(result_2[i] - result_1[i])
            list_one = tuple(list_one)
            result_list.append(list_one)
        return result_list
    
    def get_result_with_different_length(result1, result2):
    """二者长度不相同时"""
        result_both = []
        result_not_senguo_in = []
        result_all = result2
        result_list = []
        date_list = []
        for result_1 in result1:
            date_list.append(result_1[0])
        for date in date_list:
            for value in result2:
                if date in value[0]:
                    result_both.append(value)
                    result_all.remove(value)
        result_list_same = get_result_with_same_length(result1, result_both)
        result_list.extend(result_list_same)
        result_list.extend(result_all)
        return result_list
    
    if length1 == length2:
        result_list = get_result_with_same_length(result1,result2)
    
    if length1 != length2:
        result_list = get_result_with_different_length(result1,result2)

但是这样写代码过于冗长，而且非常不友好，看起来也不简洁。

后来决定将其转成字典，通过字典解决。

map_result_1 = { k[0]: list(k[1:]) for k in result1}
    map_result_2 = { k[0]: list(k[1:]) for k in result2}
    for k, v in map_result_1.items():
        if k in map_result_2:
            for i in range(len(v)):
                map_result_2[k][i] -= v[i]
    result_list = []
    for k, v in map_result_2.items():
        v.insert(0, k)
        result_list.append(v)

这里主要是学习字典的使用。