欢迎访问 生活随笔!

生活随笔

当前位置: 首页 > 编程语言 > python >内容正文

python

python:列表求差

发布时间:2024/1/18 python 35 豆豆
生活随笔 收集整理的这篇文章主要介绍了 python:列表求差 小编觉得挺不错的,现在分享给大家,帮大家做个参考.

列表格式:

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)]orresult1 = [("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 = 0list_one = []list_one.append(result_1[i])while i < len(result_1) - 1:i += 1list_one.append(result_2[i] - result_1[i])list_one = tuple(list_one)result_list.append(list_one)return result_listdef get_result_with_different_length(result1, result2): """二者长度不相同时"""result_both = []result_not_senguo_in = []result_all = result2result_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_listif 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)

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

总结

以上是生活随笔为你收集整理的python:列表求差的全部内容,希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错,欢迎将生活随笔推荐给好友。