BUG: fixes indexing with monotonic decreasing DTI (#19362)

pandas-dev · Apr 14, 2018 · 3c84724 · 3c84724
1 parent d104ecd
commit 3c84724
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/pandas/core/indexes/datetimelike.py b/pandas/core/indexes/datetimelike.py
@@ -342,7 +342,8 @@ def _format_with_header(self, header, **kwargs):
     def __contains__(self, key):
         try:
             res = self.get_loc(key)
-            return is_scalar(res) or type(res) == slice or np.any(res)
+            return is_scalar(res) or isinstance(res, slice) or \
+                (is_list_like(res) and len(res))
         except (KeyError, TypeError, ValueError):
             return False
 

diff --git a/pandas/tests/indexes/datetimes/test_datetime.py b/pandas/tests/indexes/datetimes/test_datetime.py
@@ -368,3 +368,14 @@ def test_factorize_dst(self):
     def test_unique(self, arr, expected):
         result = arr.unique()
         tm.assert_index_equal(result, expected)
+
+    def test_monotone_DTI_indexing_bug(self):
+        # GH 19362
+
+        df = pd.DataFrame({'A': [1, 2, 3]},
+                          index=pd.date_range('20170101',
+                                              periods=3)[::-1])
+        expected = pd.DataFrame({'A': 1},
+                                index=pd.date_range('20170103',
+                                                    periods=1))
+        tm.assert_frame_equal(df.loc['2017-01-03'], expected)