我有一个有序(即排序)列表,其中包含按升序排序的日期(作为日期时间对象)。在
我想编写一个函数,遍历这个列表并生成每个月的第一个可用日期的另一个列表。在
例如,假设我的排序列表包含以下数据:
A = [
'2001/01/01',
'2001/01/03',
'2001/01/05',
'2001/02/04',
'2001/02/05',
'2001/03/01',
'2001/03/02',
'2001/04/10',
'2001/04/11',
'2001/04/15',
'2001/05/07',
'2001/05/12',
'2001/07/01',
'2001/07/10',
'2002/03/01',
'2002/04/01',
]
返回的列表将是
^{pr2}$我建议的逻辑应该是这样的:
def extract_month_first_dates(input_list, start_date, end_date):
#note: start_date and end_date DEFINITELY exist in the passed in list
prev_dates, output = [],[] # <- is this even legal?
for (curr_date in input_list):
if ((curr_date < start_date) or (curr_date > end_date)):
continue
curr_month = curr_date.date.month
curr_year = curr_date.date.year
date_key = "{0}-{1}".format(curr_year, curr_month)
if (date_key in prev_dates):
continue
else:
output.append(curr_date)
prev_dates.append(date_key)
return output
有什么意见和建议吗?-这能改进成更“Python”吗?在
搜索列表是一个O(n)操作。我想你可以简单地检查一下钥匙是否是新的:
这里有一个简单的解决方案
classic python
,即没有itertools;),而且不言自明输出:
^{pr2}$相关问题 更多 >
编程相关推荐