面试题
输入一个时间段,a=202103,b=202105,
输出列表:[“2021-03”,“2021-04”,“2021-05”]
实现代码
# 输入一个时间段,a=202103,b=202105,
# 输出列表["2021-03","2021-04","2021-05"]
# 作者-上海悠悠 QQ交流群:717225969
# blog地址 https://www.cnblogs.com/yoyoketang/
def out(a=202103, b=202105):
a1 = str(a)[:4]
a2 = str(a)[4:]
b1 = str(b)[:4]
b2 = str(b)[4:]
# print(a1, a2, b1, b2)
result = []
if int(b1) == int(a1):
if int(a2) <= int(b2):
for i in range(int(a2), int(b2)+1):
result.append('%s-%02d'%(b1, i))
else:
print("起始月份不能大于结束月份")
elif int(b1) > int(a1):
for i in range(int(a1), int(b1)+1):
# 判断是起始年
if int(a1) == i:
for j in range(int(a2), 12+1):
result.append('%s-%02d'%(a1, j))
# 判断是最后一年
elif int(b1) == i:
for j in range(1, int(b2)+1):
result.append('%s-%02d'%(b1, j))
# 中间的年份
else:
for j in range(1, 12+1):
result.append('%s-%02d'%(i, j))
else:
print("起始年份不能大于结束年份")
return result
if __name__ == '__main__':
print(out())
print(out(a=202001, b=202103))
运行结果
['2021-03', '2021-04', '2021-05']
['2020-01', '2020-02', '2020-03', '2020-04', '2020-05', '2020-06', '2020-07', '2020-08', '2020-09', '2020-10', '2020-11', '2020-12', '2021-01', '2021-02', '2021-03']