From d22e816034416a1e3ba314b9629c40027a4d0905 Mon Sep 17 00:00:00 2001
From: ShaharBental <shahar.bental@gmail.com>
Date: Mon, 26 Dec 2016 13:44:19 +0200
Subject: [PATCH 1/2] GH14833: Added searchsorted to MultiIndex

Added searchsorted implementation to MultiIndex
---
 pandas/indexes/multi.py         | 20 ++++++++++++++++++++
 pandas/tests/test_multilevel.py | 12 ++++++++++++
 2 files changed, 32 insertions(+)

diff --git a/pandas/indexes/multi.py b/pandas/indexes/multi.py
index 132543e0e386c..c6c93113b5e56 100644
--- a/pandas/indexes/multi.py
+++ b/pandas/indexes/multi.py
@@ -1470,6 +1470,26 @@ def get_indexer(self, target, method=None, limit=None, tolerance=None):
 
         return _ensure_platform_int(indexer)
 
+    def searchsorted(self, value, side='left', sorter=None):
+        ans = []
+        for val in value:
+            left_index=0
+            right_index=None
+            for dim in range(0,self.labels.__len__()):
+                v=self.levels[dim][self.labels[dim]]
+                if right_index==None: right_index=len(v)
+                v=v[left_index:right_index]
+                if (dim<self.labels.__len__()-1):
+                    left_addition = v.searchsorted(val[dim],'left')
+                    right_addition = v.searchsorted(val[dim],'right')
+                    right_index=left_index+right_addition
+                    left_index=left_index+left_addition
+                else:
+                    ans.append(left_index+ v.searchsorted(val[dim],side))
+        return ans
+
+
+
     def reindex(self, target, method=None, level=None, limit=None,
                 tolerance=None):
         """
diff --git a/pandas/tests/test_multilevel.py b/pandas/tests/test_multilevel.py
index 4e7ace4173227..3f56e7ac2abab 100755
--- a/pandas/tests/test_multilevel.py
+++ b/pandas/tests/test_multilevel.py
@@ -2448,6 +2448,18 @@ def test_iloc_mi(self):
 
         assert_frame_equal(result, expected)
 
+    def test_searchsorted(self):
+        # GH 14833
+        # Test if MultiIndex searchsorted returns the right results for both left and right search sorted
+        df_mi = pd.MultiIndex([[0,1], ["a","c"]], [[0,0,1], [0,1,1]])
+        expected_left = list([0,1,3,1])
+        expected_right = list([0,1,3,2])
+
+        result_left = df_mi.searchsorted([(-1, "b"),(0,"b"),(2,"a"),(0,"c")],side='left')
+        result_right = df_mi.searchsorted([(-1, "b"),(0,"b"),(2,"a"),(0,"c")],side='right')
+
+        self.assertEqual(result_left,expected_left)
+        self.assertEqual(result_right,expected_right)
 
 if __name__ == '__main__':
     nose.runmodule(argv=[__file__, '-vvs', '-x', '--pdb', '--pdb-failure'],

From de97bdb4c079f9c2d4e779a0585ed76174c51ff6 Mon Sep 17 00:00:00 2001
From: ShaharBental <shahar.bental@gmail.com>
Date: Mon, 26 Dec 2016 20:28:20 +0200
Subject: [PATCH 2/2] GH14833: changed implementation to simply raise a
 NotImplementedException

---
 pandas/indexes/multi.py         | 19 +------------------
 pandas/tests/test_multilevel.py | 15 +++++----------
 2 files changed, 6 insertions(+), 28 deletions(-)

diff --git a/pandas/indexes/multi.py b/pandas/indexes/multi.py
index c6c93113b5e56..1b1c30b8d0d6b 100644
--- a/pandas/indexes/multi.py
+++ b/pandas/indexes/multi.py
@@ -1471,24 +1471,7 @@ def get_indexer(self, target, method=None, limit=None, tolerance=None):
         return _ensure_platform_int(indexer)
 
     def searchsorted(self, value, side='left', sorter=None):
-        ans = []
-        for val in value:
-            left_index=0
-            right_index=None
-            for dim in range(0,self.labels.__len__()):
-                v=self.levels[dim][self.labels[dim]]
-                if right_index==None: right_index=len(v)
-                v=v[left_index:right_index]
-                if (dim<self.labels.__len__()-1):
-                    left_addition = v.searchsorted(val[dim],'left')
-                    right_addition = v.searchsorted(val[dim],'right')
-                    right_index=left_index+right_addition
-                    left_index=left_index+left_addition
-                else:
-                    ans.append(left_index+ v.searchsorted(val[dim],side))
-        return ans
-
-
+        raise NotImplementedError()
 
     def reindex(self, target, method=None, level=None, limit=None,
                 tolerance=None):
diff --git a/pandas/tests/test_multilevel.py b/pandas/tests/test_multilevel.py
index 3f56e7ac2abab..33f266954c767 100755
--- a/pandas/tests/test_multilevel.py
+++ b/pandas/tests/test_multilevel.py
@@ -2450,16 +2450,11 @@ def test_iloc_mi(self):
 
     def test_searchsorted(self):
         # GH 14833
-        # Test if MultiIndex searchsorted returns the right results for both left and right search sorted
-        df_mi = pd.MultiIndex([[0,1], ["a","c"]], [[0,0,1], [0,1,1]])
-        expected_left = list([0,1,3,1])
-        expected_right = list([0,1,3,2])
-
-        result_left = df_mi.searchsorted([(-1, "b"),(0,"b"),(2,"a"),(0,"c")],side='left')
-        result_right = df_mi.searchsorted([(-1, "b"),(0,"b"),(2,"a"),(0,"c")],side='right')
-
-        self.assertEqual(result_left,expected_left)
-        self.assertEqual(result_right,expected_right)
+        # Test if MultiIndex searchsorted raises the NotImplementedException
+        df_mi = pd.MultiIndex([[0], ["a"]], [[0], [0]])
+        with self.assertRaises(Exception) as context:
+            df_mi.searchsorted((1, "b"))
+        self.assertTrue(context.exception,NotImplementedError())
 
 if __name__ == '__main__':
     nose.runmodule(argv=[__file__, '-vvs', '-x', '--pdb', '--pdb-failure'],