ITPub博客

首页 > 大数据 > 可视化 > python返回特定时间

python返回特定时间

原创 可视化 作者:2008081036 时间:2017-11-09 13:29:22 0 删除 编辑

返回特定时间

下方的 which_date() 函数实现返回某一起始时间后特定一段时间的日期。特定的一段时间有两种不同的格式:一种是不同的天数,如 "1 day" 或 "30 days",另一种是不同的星期数,如 "2 weeks" 或 "12 weeks"。


from datetime import datetime, timedelta
def which_date(start_date,time):
    end_date=datetime.strptime(start_date,"%Y/%m/%d")
    n=int(time.split()[0])
    t=time.split()[1]   
    if t in 'days':
        end_date=end_date+timedelta(days=n)
    elif t in 'weeks':
        end_date=end_date+timedelta(weeks=n)
    end_date=end_date.strftime("%Y/%m/%d")
    return end_date

def test():
    assert which_date('2016/02/10','35 days') == '2016/03/16'
    assert which_date('2016/12/21','3 weeks') == '2017/01/11'
    assert which_date('2015/01/17','1 week') == '2015/01/24'
    print("All tests completed.")
    
if __name__ == "__main__":
    test()

来自 “ ITPUB博客 ” ,链接:http://blog.itpub.net/25465866/viewspace-2147028/,如需转载,请注明出处,否则将追究法律责任。

请登录后发表评论 登录
全部评论

注册时间:2014-05-05

  • 博文量
    26
  • 访问量
    88541