Python求前后第n个月函数

岚天逸剑

于 2023-05-25 22:23:17 发布

阅读量201

点赞数

CC 4.0 BY-SA版权

分类专栏： python 文章标签： python 开发语言

本文链接：https://blog.youkuaiyun.com/Aquester/article/details/130876699

python 专栏收录该内容

3 篇文章

订阅专栏

该文章提供了两个Python函数，next_month和prev_month，分别用于计算给定日期之后和之前的n个月份。这两个函数通过处理年份和月份，实现了对日期的增减操作，适用于数据分析中的日期处理场景。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

数据分析常用到

# 求往后第 n 个月
# monthstr 格式：YYYY-MM
#
# print next_month("2023-05", 1)
# 2023-06
# print next_month("2023-05", 6)
# 2023-11
# print next_month("2023-05", 8)
# 2024-01
def next_month(monthstr, n):
    yearnum = int(monthstr[0:4])
    monthnum = int(monthstr[5:7])
    date = datetime.datetime.strptime(monthstr + "-01", '%Y-%m-%d')    
    for i in range(0, n):
        if monthnum == 12:
            yearnum += 1
            monthnum = 1
        else:
            monthnum += 1
    return "{}-{:02}".format(yearnum, monthnum)

# 求往前第 n 个月
# monthstr 格式：YYYY-MM
#
# print prev_month("2023-05", 1)
# 2023-04
# print prev_month("2023-05", 3)
# 2023-02
# print prev_month("2023-05", 5)
# 2022-12
def prev_month(monthstr, n):
    yearnum = int(monthstr[0:4])
    monthnum = int(monthstr[5:7])
    date = datetime.datetime.strptime(monthstr + "-01", '%Y-%m-%d')    
    for i in range(0, n):
        if monthnum == 1:
            yearnum -= 1
            monthnum = 12
        else:
            monthnum -= 1
    return "{}-{:02}".format(yearnum, monthnum)