如何在Python中使用循环时根据条件合并两个大列表

Question

我有两个相当大的列表，分别命名为 A 和 B，它们就像

['[20130602, 20130608]',
 '[20130609, 20130615]',
 '[20130630, 20130706]',
 '[20130804, 20130810]',
 '[20130901, 20130907]',
 '[20140601, 20140607]',
 '[20140720, 20140726]',
 '[20140727, 20140802]',
 '[20140803, 20140809]',
 '[20140810, 20140816]',
...
 '[20200719, 20200725]',
 '[20200726, 20200801]',
 '[20200802, 20200808]',
 '[20200809, 20200815]',
 '[20200816, 20200822]',
 '[20200823, 20200829]',
 '[20200830, 20200905]',
 '[20200906, 20200912]']

我

['[20131201, 20131207]',
 '[20131208, 20131214]',
 '[20131229, 20140104]',
 '[20140105, 20140111]',
 '[20140112, 20140118]',
 '[20140119, 20140125]',
 '[20141207, 20141213]',
 '[20141228, 20150103]',
 '[20150104, 20150110]',
...
 '[20210103, 20210109]',
 '[20210124, 20210130]',
 '[20210131, 20210206]',
 '[20210207, 20210213]',
 '[20210214, 20210220]',
 '[20210221, 20210227]']

想按日期顺序合并这两个列表。 理想的结果是，

'[20130602, 20130608]',
'[20130609, 20130615]',
'[20130630, 20130706]',
'[20130804, 20130810]',
'[20130901, 20130907]', ---> summer in 2013
'[20131201, 20131207]',
'[20131208, 20131214]',
'[20131229, 20140104]',
'[20140105, 20140111]',
'[20140112, 20140118]',
'[20140119, 20140125]', ---> winter in the late of 2013 and in the early of 2014
'[20140601, 20140607]',
'[20140720, 20140726]',
'[20140727, 20140802]',
'[20140803, 20140809]', ... ] till the end of the last day in the early of 2021 from the second list. 

首先，我考虑使用Python中的while、for和if等循环。

y = 13
while y < 22:
    for i in range(len(windS)):
        if windS[i][3:5] == str(y):
            count.append(windS[i])
    for j in range(len(cindS)):
        if cindS[j][3:5] == str(y):
            count.append(cindS[j])
        elif cindS[j][3:5] == str(y+1) and cindS[j][3:5] == '01' or '02' or '03':
            count.append(cindS[j])
            y += 1

但它带来了令人不安的结果，比如 2013 年只有一个夏季，随后所有年份都按顺序排列冬季。 我怎样才能得到理想的结果？

Answer 1

我认为，下面的代码可以解决您的问题。

date_list1 = ['[20130602, 20130608]',
 '[20130609, 20130615]',
 '[20130630, 20130706]',
 '[20130804, 20130810]',
 '[20130901, 20130907]',
 '[20140601, 20140607]',
 '[20140720, 20140726]',
 '[20140727, 20140802]',
 '[20140803, 20140809]',
 '[20140810, 20140816]',
 '[20200719, 20200725]',
 '[20200726, 20200801]',
 '[20200802, 20200808]',
 '[20200809, 20200815]',
 '[20200816, 20200822]',
 '[20200823, 20200829]',
 '[20200830, 20200905]',
 '[20200906, 20200912]']

date_list2 = ['[20131201, 20131207]',
 '[20131208, 20131214]',
 '[20131229, 20140104]',
 '[20140105, 20140111]',
 '[20140112, 20140118]',
 '[20140119, 20140125]',
 '[20141207, 20141213]',
 '[20141228, 20150103]',
 '[20150104, 20150110]',
 '[20210103, 20210109]',
 '[20210124, 20210130]',
 '[20210131, 20210206]',
 '[20210207, 20210213]',
 '[20210214, 20210220]',
 '[20210221, 20210227]']

len_date_list_1 = len(date_list1)
len_date_list_2 = len(date_list2)
  
res = []
i, j = 0, 0
  
while i < len_date_list_1 and j < len_date_list_2:
    if date_list1[i] < date_list2[j]:
      res.append(date_list1[i])
      i += 1
  
    else:
      res.append(date_list2[j])
      j += 1
  
res = res + date_list1[i:] + date_list2[j:]

print(res)

Answer 2

我将首先组合/合并列表，然后对它们进行排序。我认为同时进行这两项工作会更复杂

Answer 3

Python 标准库中有一个函数可以合并两个排序列表： <代码>heapq.merge。
您可以在您的情况下使用它，如下所示：

from heapq import merge

count = list(merge(windS, cindS, key=lambda x: x[1:9]))

请注意， list 调用是必要的，因为 merge 返回一个迭代器，并且您可能需要一个列表。

key=lambda x: x[1:9] 部分表示：按照以下条件合并：列表中每个项目的字母 2 到 9（索引从零开始）。

Answer 4

不知道为什么这些列表包含字符串，这些列表本身显然是字符串化的列表。

在这种情况下，最简单的解决方案是

result = sorted(windS + cindS)

将两个列表连接成一个大列表，并按词法对结果进行排序。这似乎适用于上面的示例数据。