BUG: preserve index names in MultiIndex.drop, close #1513

pandas-dev · Jun 22, 2012 · 836ecab · 836ecab
1 parent 44a8295
commit 836ecab
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/pandas/core/index.py b/pandas/core/index.py
@@ -2295,7 +2295,8 @@ def delete(self, loc):
         new_index : MultiIndex
         """
         new_labels = [np.delete(lab, loc) for lab in self.labels]
-        return MultiIndex(levels=self.levels, labels=new_labels)
+        return MultiIndex(levels=self.levels, labels=new_labels,
+                          names=self.names)
 
     get_major_bounds = slice_locs
 

diff --git a/pandas/tests/test_multilevel.py b/pandas/tests/test_multilevel.py
@@ -1355,6 +1355,16 @@ def test_drop_level(self):
         expected = self.frame.ix[[0, 2, 3, 6, 7, 9]].T
         assert_frame_equal(result, expected)
 
+    def test_drop_preserve_names(self):
+        index = MultiIndex.from_arrays([[0, 0, 0, 1, 1, 1],
+                                        [1, 2, 3, 1, 2, 3]],
+                                       names=['one', 'two'])
+
+        df = DataFrame(np.random.randn(6, 3), index=index)
+
+        result = df.drop([(0, 2)])
+        self.assert_(result.index.names == ['one', 'two'])
+
     def test_unicode_repr_issues(self):
         levels = [Index([u'a/\u03c3', u'b/\u03c3',u'c/\u03c3']),
                   Index([0, 1])]