Как прибавить несколько месяцев с помощью datetime?

Question

[PesyCorm]

Привет! Есть задачка, нет решения. Условие:

Написать функцию change_month, которая к переданной дате прибавляет/вычитает переданное кол-во месяцев.
Например:change_month(’12.12.12’, 7) -> ’12.07.13’ change_month(’01.11.10’, -5) -> ’01.06.10’
Разрешено использовать только внутренние библиотеки питона, желательно ограничиться datetime

Есть вариант такого исполнения:

>>> def change_month(mydate, months):
        skip_months = datetime.datetime(mydate.year + int(months / 12), (mydate.month % 12+ months % 12), 1)
        skip_months = skip_months + datetime.timedelta(days=mydate.day)
        print(skip_months)

>>> change_month(datetime.datetime(2021, 1, 1), 12)
2022-01-02 00:00:00

Но если (в случае данного решения), например, к дате 01.01.2021 добавить 12 месяцев ( == 1 год), то получим дату 02.01.2021 число, что не есть правда.
Как быть?
Спасибо за помощь!

p.s.: Еще раз уточню, что пользоваться можно только встроенными библиотеками

Answer 1

[soremix]

Почему не правильно?

Формат даты представлен в виде ГОД-МЕСЯЦ-ДЕНЬ.
Изначально вы отправили дату 2021-01-01 00:00:00

Сделав так

skip_months = datetime.datetime(mydate.year + int(months / 12), (mydate.month % 12+ months % 12), 1)

вы получили 2022-01-01 00:00:00. То есть, 2022 год, 1 января. Потом к этому результату прибавили зачем-то еще один день (

skip_months = skip_months + datetime.timedelta(days=mydate.day)

)
И в результате получили 2022-01-02 00:00:00 - 2022 год, 2 января

Comments

[PesyCorm]

Точно! Поправил второй оператор

skip_months = skip_months + datetime.timedelta(days=(mydate.day-1))

Затестил дату и сразу словил новую ошибку

>>> change_month(datetime.datetime(2021, 2, 2), 11)
Traceback (most recent call last):
  File "<pyshell#11>", line 1, in <module>
    change_month(datetime.datetime(2021, 2, 2), 11)
  File "<pyshell#5>", line 2, in change_month
    skip_months = datetime.datetime(mydate.year + int(months / 12), (mydate.month % 12+ months % 12), 1)
ValueError: month must be in 1..12

Когда прибавляю ко 2-ому месяцу еще 11, получаю выход за границы. Лучший способ был бы через дельту, но она не работает с месяцами.

[soremix]

PesyCorm, нужно сначала сложить текущее кол-во месяцев с кол-вом для пропуска, и затем получать остаток от деления на 12. И для правильного указания года нужно сделать то же самое, затем разделить нацело на 12.

months_delta = (mydate.month + months) // 12
skip_months = datetime.datetime(mydate.year + months_delta, (mydate.month+months)%12, 1)

да и добавить проверку бы на сумму, чтобы если сумма месяцев была равна 12, чтоб не вычислялся остаток от деления

def change_month(mydate, months):
    months_delta = (mydate.month + months) // 12
    if mydate.month+months == 12:
        skip_months = datetime.datetime(mydate.year + months_delta, 12, 1)
    else:
        skip_months = datetime.datetime(mydate.year + months_delta, (mydate.month+months)%12, 1)
    print(skip_months)

Хотя я бы таким не страдал и использовал relativedelta, но раз надо то ладно

[PesyCorm]

soremix, Спасибо!

Answer 2

[dmshar]

Ни непосредственно месяца, ни непосредственно года задать с помощью timedelta нельзя, и это понятно. Год может быть 365 и 366 дней, а месяц - 28,29,30 и 31.
Тем не менее обойти это можно. Ниже один варинат на чистом datatime, а остальные - с привлечением других библиотек.
Показываю на примере года. Думаю, с месяцем разберетесь сами по аналогии

import datetime as dt
from dateutil.relativedelta import relativedelta
import pandas as pd
d1=dt.datetime(2021, 5, 16)
print ('d1==',d1)
d2=d1.replace(year=d1.year+1)
print ('d2==',d2)
d3=d1+relativedelta(years=1)
print ('d3==',d3)
d4=d1+pd.DateOffset(years=1)
print ('d4==',d4)

Результаты:

d1== 2021-05-16 00:00:00
d2== 2022-05-16 00:00:00
d3== 2022-05-16 00:00:00
d4== 2022-05-16 00:00:00

Comments

[PesyCorm]

Спасибо!