以下是一个示例代码,用于将员工连续工作的天数进行分组:
from itertools import groupby
def group_consecutive_days(dates):
groups = []
for k, g in groupby(enumerate(dates), lambda i_x:i_x[0]-i_x[1]):
group = list(map(lambda x: x[1], g))
groups.append(group)
return groups
# 示例输入
dates = [1, 2, 3, 5, 6, 7, 9, 10, 11, 15, 16, 17]
groups = group_consecutive_days(dates)
# 输出分组结果
for group in groups:
print(group)
输出结果:
[1, 2, 3]
[5, 6, 7]
[9, 10, 11]
[15, 16, 17]
这段代码使用了itertools.groupby
函数来对连续的天数进行分组。首先通过enumerate
函数将日期列表转换为(索引, 日期)
的元组列表,然后通过lambda
函数指定分组条件,即索引与日期的差值。groupby
函数根据这个差值进行分组,返回一个迭代器,迭代器的每个元素都是(差值, 具有相同差值的元素组成的迭代器)
的元组。最后,使用map
函数将每个迭代器转换为包含日期的列表,并将这些列表添加到groups
列表中。最终,返回groups
列表,即分好组的连续天数。
上一篇:不同源的自动iframe高度