datetime函数，用于将yyyymm输入转换为月初和月底日期输出

Question

我不习惯使用datetime

来自yyyymm（int）中的输入

input = 202003

我想创建一个函数，将其转换为该月第一天和最后一天的int变量yyyymmdd，在本例中：

start = 20200301
end =  20200331

谢谢

Answer 1

代码：

from datetime import datetime, date, timedelta

string = "202003" # > Your input string
start = datetime.strptime(string, "%Y%m").date()

nextmonth = 1 if start.month == 12 else start.month + 1 #  > if current month is 12 cycle to 1
year = start.year + 1 if nextmonth == 1 else start.year

end = date(year, nextmonth, 1) - timedelta(days=1)

print("start:", start)
print("end:", end)

例如：

如果输入string为202003，则输出为

start: 2020-03-01
end: 2020-03-31

如果输入string为202002，则输出为

start: 2020-02-01
end: 2020-02-29

如果输入string为202012，则输出为

start: 2020-12-01
end: 2020-12-31