将连续日期分组在一起

我在 Excel/csv 中有一个请病假的（许多）员工列表，按以下格式列出。每个病假实例都有自己的线路。我想添加另一列“结果”，它记录病期的长度。例如，Mon-Tues-Wed 表示这三个条目中的每一个都标有 3。

我是Python新手，我想知道这种方法是否理想，尽管我看不出SQL会变得更容易，除了为每个员工创建表（简单）然后对其进行分析（困难）

我的目标是能够将 1 天长的时间段与 10 天以上的时间段区分开。周末的奖励积分。

Person    Date       Result

A       02/04/2012     5

B       02/04/2012     2

A       03/04/2012     5

B       03/04/2012     2

A       04/04/2012     5

A       05/04/2012     5

A       06/04/2012     5

B       25/04/2012     1

A       25/04/2012     2

A       26/04/2012     2

B       30/04/2012     1

def group(iterable):
    myIter = iter(iterable)

    run = [next(myIter)]
    def continuesRun(x):
        return run[-1]==x-1

    for x in myIter:
        if continuesRun(x):
            run.append(x)
        else:
            yield run
            run = [x]
    yield run

Demo:

>>> list( group([1,10,11,12,20,21]) )
[[1], [10, 11, 12], [20, 21]]

要将其应用到您的情况，请定义函数continuesRun像这样，用伪代码：

def continuesRun(date):
    previousDate = run[-1]
    return previousDate==date-1day or (previousDate.weekday==Friday and previousDate==date-3day)

旁注：在我个人看来，将周末附近的病假时间计算为可能会多出 2 或 4 天，这在道德/务实上似乎有点错误。但如果你有充分的理由这样做，我又有什么资格去评判呢？ =) 要计算这些，请对您的跑步进行后处理：如果第一天是星期一，则添加 2；如果最后一天是星期五，则添加 2，然后添加len(d for d in range(run[-1]-run[0]) if (run[0]+d*day).isWeekend())。当然这不包括假期，在这种情况下你会这样做.isHoliday() or .isWeekend()并使“add 2”逻辑与len(...)逻辑，通过迭代直到找到非假期，并在与跑步相邻的每个假期或周末惩罚该人。