Python find index of all dates between two dates

Question

I am trying to implement equivalent of numpy.where for dates as follows:

from datetime import date, timedelta as td, datetime
d1 = datetime.strptime('1/1/1995', "%m/%d/%Y")
d2 = datetime.strptime('12/31/2015', "%m/%d/%Y")

AllDays = []
while(d1<=d2):
    AllDays.append(d1)
    d1 = d1 + td(days=1)

validDate = AllDays
trainStDt = '1/1/1995'
trainEnDt = '12/31/2013'
testStDt = '1/1/2014'
testEnDt = '12/31/2015'

indTrain = (validDate >= datetime.strptime(trainStDt,'%m/%d/%Y')) & (validDate <=
                                                                           datetime.strptime(trainEnDt,'%m/%d/%Y'))
indTest = (validDate >= datetime.strptime(testStDt,'%m/%d/%Y')) & (validDate <=
                                                                          datetime.strptime(testEnDt,'%m/%d/%Y'))
trainDates = validDate[indTrain]
testDates = validDate[indTest]

print trainDates[0]
print trainDates[-1:]
print testDates[0]
print testDates[-1:]

However: (1) indTrain doesn't work as it is trying to compare list to datetime (2) my solution is to loop through each element of validDates

Is there a better way to do it?

farenorth · Accepted Answer · 2016-05-28 16:03:51Z

1

Just turn your list into an array. Add import numpy as np to the top of your script, and after your while loop, add:

AllDays = np.array(AllDays)

edited May 28, 2016 at 16:03

answered May 28, 2016 at 14:56

farenorth

11k2 gold badges45 silver badges46 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Python find index of all dates between two dates

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related