python - 在 Python 中创建连续年份月份的列表

Question

我尝试使用以下命令创建从 2020-06(上个月从今天开始)到 2021-05 的年-月列表以下代码:

from datetime import datetime
from dateutil.relativedelta import relativedelta
need_months = list(range(-1, 12))
print([(datetime.today()+ relativedelta(months=i if i < 0 else i - 1)).strftime('%Y-%m') for i in need_months])

输出:

['2020-06', '2020-06', '2020-07', '2020-08', '2020-09', '2020-10', '2020-11', '2020-12', '2021-01', '2021-02', '2021-03', '2021-04', '2021-05']

但您可能会注意到，列表中有两个 2020-06。

我怎样才能正确地做到这一点？

预期结果:

['2020-06', '2020-07', '2020-08', '2020-09', '2020-10', '2020-11', '2020-12', '2021-01', '2021-02', '2021-03', '2021-04', '2021-05']

Answer 1

这里是可能的用途period_range将今天转换为月份期间 Timestamp.to_period ，对于上个月减去 1 并添加 12 个周期参数:

now = pd.Timestamp.now().to_period('m')
L = pd.period_range(now-1, freq='M', periods=12).strftime('%Y-%m').tolist()
print (L)
['2020-06', '2020-07', '2020-08', '2020-09', '2020-10', '2020-11', 
 '2020-12', '2021-01', '2021-02', '2021-03', '2021-04', '2021-05']