python:列表求差
生活随笔
收集整理的這篇文章主要介紹了
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:列表求差的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 排列组合(两种方法)
- 下一篇: linux使用管道显示某一进程的运行情况