Split the intersection into two sub-functions

pandas-dev · jreback · Jun 6, 2019 · Apr 21, 2019 · Apr 27, 2019 · Apr 27, 2019
commit 7cb7d2cf9f1aed2e449ef937b63b0d0212f6f7c5
diff --git a/pandas/core/indexes/interval.py b/pandas/core/indexes/interval.py
@@ -1102,25 +1102,35 @@ def intersection(self, other, sort=False):
             raise TypeError(msg)
 
         if self.left.is_unique and self.right.is_unique:
-            lindexer = self.left.get_indexer(other.left)
-            rindexer = self.right.get_indexer(other.right)
-            match = (lindexer == rindexer) & (lindexer != -1)
-            indexer = lindexer.take(match.nonzero()[0])
-            taken = self.take(indexer)
+            taken = self._intersection_unique(other)
         else:
             # duplicates
-            lmiss = other.left.get_indexer_non_unique(self.left)[1]
-            rmiss = other.right.get_indexer_non_unique(self.right)[1]
-            import functools
-            indexer = functools.reduce(np.setdiff1d, (np.arange(len(self)),
-                                                      lmiss, rmiss))
-            taken = self[indexer]
+            taken = self._intersection_non_unique(other)
 
         if sort is None:
             taken = taken.sort_values()
 
         return taken
 
+    def _intersection_unique(self, other):
+        lindexer = self.left.get_indexer(other.left)
+        rindexer = self.right.get_indexer(other.right)
+
+        match = (lindexer == rindexer) & (lindexer != -1)
+        indexer = lindexer.take(match.nonzero()[0])
+
+        return self.take(indexer)
+
+    def _intersection_non_unique(self, other):
+        lmiss = other.left.get_indexer_non_unique(self.left)[1]
+        rmiss = other.right.get_indexer_non_unique(self.right)[1]
+
+        import functools
+        indexer = functools.reduce(np.setdiff1d, (np.arange(len(self)),
+                                                  lmiss, rmiss))
+
+        return self[indexer]
+
     def _setop(op_name, sort=None):
         def func(self, other, sort=sort):
             other = self._as_like_interval_index(other)