PERF: Use scalar lookups for TradingCalendar.schedule

When retrieving the open and close for a given session, we only care
about the scalar values, so using DataFrame.at instead of DataFrame.loc
is significantly faster.
This commit is contained in:
Andrew Daniels
2017-04-10 17:23:07 -04:00
parent bd1b7f263c
commit 6dd1616c15
2 changed files with 15 additions and 8 deletions
+4 -2
View File
@@ -452,8 +452,10 @@ class ExchangeCalendarTestBase(object):
(2, 1),
])
def test_minute_index_to_session_labels(self, interval, offset):
minutes = self.calendar.minutes_for_sessions_in_range('2011-01-04',
'2011-04-04')
minutes = self.calendar.minutes_for_sessions_in_range(
pd.Timestamp('2011-01-04', tz='UTC'),
pd.Timestamp('2011-04-04', tz='UTC'),
)
minutes = minutes[range(offset, len(minutes), interval)]
np.testing.assert_array_equal(