Создание функции IsHoliday в фрейме данных панд - PullRequest
0 голосов
/ 12 декабря 2018

Я пытаюсь создать функцию IsHoliday в моем файле pd.dataframe с указанием даты и времени в качестве индекса, основанного на файле csv, который включает выходные дни в одном году.Имея небольшой опыт работы с пандами, я могу придумать итеративный подход, сравнивая значения двух фреймов данных. Чтобы быть более конкретным:

for i in range(0,len(Holidays)-1):
 for j in range(0,len(df)-1):
    if (Holidays.loc[i,'month']==df.loc[j,'month'] and 
      Holidays.loc[i,'day']==df.loc[j,'day'] ):
        df.loc[j,'Isholiday']=1

    else: 
        df.loc[j,'Isholiday']=0

Мой вопрос заключается в том, как это можно реализовать способом панд, избегая всего времени обработки.

1 Ответ

0 голосов
/ 12 декабря 2018

Вы можете использовать np.where для этого:

df['IsHoliday'] = np.where(df.index.isin(Holidays),True,False)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...