您现在的位置是：首页 > Python

当前栏目

python：列表求差

Python

2023-03-02 11:02:52 时间

列表格式：

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)

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

猜你喜欢

最长无重复子串
写技术博客的一些心得分享
Java 多线程（七）：线程池
Java 多线程（五）：锁（三）
Java 多线程（四）：锁（二）
Java 多线程（三）：锁（一）
Java 多线程（二）：并发编程的三大特性
线性时间非比较类排序
Java 多线程（一）：基础
合并k个已排序的链表
HDFS 高可用分布式环境搭建
合并两个有序数组
连续子数组的最大和
HDFS 分布式环境搭建
容器盛水问题
大数加法
HDFS 伪分布式环境搭建
设计LRU缓存结构
两数之和
使用单调栈来解决的一些问题

zl程序教程

当前栏目

python：列表求差

相关文章