Присвойте дату из диапазона l oop новому столбцу, который соответствует году из другого столбца datetime - PullRequest
0 голосов
/ 02 февраля 2020

Я использую ниже l oop, чтобы определить этот список дат. Теперь я хотел бы взять дату в этом списке и связать ее со столбцом pandas datetime дат начала работы сотрудника, отметив дату из этого списка в новом столбце, когда год здесь (i) совпадает с датой начала сотрудника год. Как мне связать два?

import datetime as dt

def get_year_day(year,month=8,day_of_week=6,num_in_month=3):
    ## set up possible ranges
    range_1 = 7*(num_in_month-1)+1
    range_2 = 7*num_in_month+1
    ## loop through possible range in the year and the month
    for i in range(range_1,range_2):
        date = dt.datetime(year=year,month=month, day=i)
        ## if we have our weekday we can break
        if date.weekday()==day_of_week:
            break
    return date

for i in range(2015,2021):
    print(i,get_year_day(i))

2015 2015-08-16 00:00:00
2016 2016-08-21 00:00:00
2017 2017-08-20 00:00:00
2018 2018-08-19 00:00:00
2019 2019-08-18 00:00:00
2020 2020-08-16 00:00:00
...