Укажите, находится ли дата-время строки в диапазоне дат - PullRequest
Новая
       7

Укажите, находится ли дата-время строки в диапазоне дат

0 голосов
/ 28 июня 2018

Я пытаюсь получить фиктивные переменные для праздников в наборе данных. У меня есть несколько диапазонов дат (pd.daterange()) с выходными и рамкой данных, к которой я хотел бы добавить фиктивный элемент, чтобы указать, находится ли дата-время в этой строке в определенном диапазоне дат указанных выходных.

Маленький пример: 

ChristmasBreak = list(pd.date_range('2014-12-20','2015-01-04').date)

dates = pd.date_range('2015-01-03', '2015-01-06, freq='H')
d = {'Date': dates, 'Number': np.rand(len(dates))}

df = pd.DataFrame(data=d)
df.set_index('Date', inplace=True)

for i, row in df.iterrows():
    if i in ChristmasBreak:
        df[i,'Christmas] = 1

if loop никогда не вводится, поэтому сопоставление дат не будет работать. Есть какой-либо способ сделать это? В этом случае также приветствуются альтернативные методы для манекенов!

Ответы [ 2 ]

0 голосов
/ 28 июня 2018

Это должно делать то, что вы хотите: 

df['Christmas'] = df.index.isin(ChristmasBreak).astype(int)
0 голосов
/ 28 июня 2018

Сначала не используйте iterrows , потому что очень медленно .

Лучше использовать dt.date с Series,isin, последнее преобразование логической маски в целое число - True s 1: 

df = pd.DataFrame(data=d)

df['Christmas'] = df['Date'].dt.date.isin(ChristmasBreak).astype(int)

Или используйте between: 

df['Christmas'] = df['Date'].between('2014-12-20', '2015-01-04').astype(int)

Если хотите сравнить с DatetimeIndex: 

df = pd.DataFrame(data=d)
df.set_index('Date', inplace=True)

df['Christmas'] = df.index.date.isin(ChristmasBreak).astype(int)

df['Christmas'] = ((df.index > '2014-12-20') & (df.index < '2015-01-04')).astype(int)

Sample 

ChristmasBreak = pd.date_range('2014-12-20','2015-01-04').date

dates = pd.date_range('2014-12-19 20:00', '2014-12-20 05:00', freq='H')
d = {'Date': dates, 'Number': np.random.randint(10, size=len(dates))}

df = pd.DataFrame(data=d)

df['Christmas'] = df['Date'].dt.date.isin(ChristmasBreak).astype(int)
print (df)
                 Date  Number  Christmas
0 2014-12-19 20:00:00       6          0
1 2014-12-19 21:00:00       7          0
2 2014-12-19 22:00:00       0          0
3 2014-12-19 23:00:00       9          0
4 2014-12-20 00:00:00       1          1
5 2014-12-20 01:00:00       3          1
6 2014-12-20 02:00:00       1          1
7 2014-12-20 03:00:00       8          1
8 2014-12-20 04:00:00       2          1
9 2014-12-20 05:00:00       1          1
...