python - 将彼此时间范围内的任意日期对象组合在一起

Question

我想将日历分成两周的间隔，从 2008-May-5 或任意起点开始。

所以我从几个日期对象开始:

import datetime as DT

raw = ("2010-08-01",
       "2010-06-25",
       "2010-07-01",
       "2010-07-08")

transactions = [(DT.datetime.strptime(datestring, "%Y-%m-%d").date(),
                 "Some data here") for datestring in raw]
transactions.sort()

通过手动分析日期，我完全能够找出哪些日期属于相同的两周间隔。我想要得到类似于这个的分组:

# Fortnight interval 1
(datetime.date(2010, 6, 25), 'Some data here')
(datetime.date(2010, 7, 1), 'Some data here')
(datetime.date(2010, 7, 8), 'Some data here')

# Fortnight interval 2
(datetime.date(2010, 8, 1), 'Some data here')

Answer 1

import datetime as DT
import itertools

start_date=DT.date(2008,5,5)

def mkdate(datestring):
    return DT.datetime.strptime(datestring, "%Y-%m-%d").date()

def fortnight(date):
    return (date-start_date).days //14

raw = ("2010-08-01",
       "2010-06-25",
       "2010-07-01",
       "2010-07-08")
transactions=[(date,"Some data") for date in map(mkdate,raw)]
transactions.sort(key=lambda (date,data):date)

for key,grp in itertools.groupby(transactions,key=lambda (date,data):fortnight(date)):
    print(key,list(grp))

产量

# (55, [(datetime.date(2010, 6, 25), 'Some data')])
# (56, [(datetime.date(2010, 7, 1), 'Some data'), (datetime.date(2010, 7, 8), 'Some data')])
# (58, [(datetime.date(2010, 8, 1), 'Some data')])

请注意，2010-6-25 是 2008-5-5 的第 55 个两周，而 2010-7-1 是第 56 个。如果您希望将它们组合在一起，只需更改 start_date(类似于 2008-5-16)。

附言。上面用到的关键工具是itertools.groupby，详细解释here .

编辑:lambda 只是一种生成 "anonymous" functions 的方法. (它们是匿名的，因为它们没有像 def 定义的函数那样被赋予名称)。在任何你看到 lambda 的地方，也可以使用 def 来创建等效函数。例如，您可以这样做:

import operator
transactions.sort(key=operator.itemgetter(0))

def transaction_fortnight(transaction):
    date,data=transaction
    return fortnight(date)

for key,grp in itertools.groupby(transactions,key=transaction_fortnight):
    print(key,list(grp))