python 通过输入来获取到指定的年月

python利用time和datetime模块获取当前月份的前N个月和后N个月

首先获取当前时间戳，将当前时间戳转化为YYYY 、MM，转化为相应的月份之后再将时间戳转化为月份

import time
import datetime

def get_YYYYMM(mon):
    '''
    I:mon   0：默认为上个月账期
            1：当月账期
            -2：当前月的三个月前的账期
    :return: YYYYMM
    '''
    mon = mon-2
    YYYYMM = time.strftime("%Y%m", time.localtime())
    YYYY = int(YYYYMM[0:4])
    MM = int(YYYYMM[4:6])
    yearadd = (MM + mon) // 12
    mon = (MM + mon) % 12 +1
    year = YYYY + yearadd
    times = int(time.mktime(datetime.date(year, mon, 1).timetuple()))
    times = time.localtime(times)
    timeYM = time.strftime('%Y%m',times)
    return timeYM
if __name__ == '__main__':
    print(get_YYYYMM(1))#获取当月账期
    print(get_YYYYMM(-1)) #获取两个月前账期
    pass