python 日历_Python获得当前日历季的第一个和最后一天

I've built a function to get the first and last day of the current quarter but it's a bit long winded. I was wondering, is there a more succinct way of accomplishing this?

I understand that pandas has a QuarterBegin() function, but I couldn't implement it in a more concise way.

import datetime as dt

from dateutil.relativedelta import relativedelta

def get_q(first=None,last=None):

today = dt.date.today()

qmonth = [1, 4, 7, 10]

if first:

for i,v in enumerate(qmonth):

if (today.month-1)//3 == i:

return dt.date(today.year,qmonth[i],1).strftime("%Y-%m-%d")

if last:

firstday = dt.datetime.strptime(get_q(first=True),"%Y-%m-%d")

lastday = firstday + relativedelta(months=3, days=-1)

return lastday.strftime("%Y-%m-%d")

EDIT: Please let me know if this would be better suited to Code Review

解决方案

You can do it this way:

import bisect

import datetime as dt

def get_quarter_begin():

today = dt.date.today()

qbegins = [dt.date(today.year, month, 1) for month in (1,4,7,10)]

idx = bisect.bisect(qbegins, today)

return str(qbegins[idx-1])

This solves the "first" case; I'm leaving the "last" case as an exercise but I suggest keeping it as an independent function for clarity (with your original version it's pretty strange what happens if no arguments are passed!).