Fix sort_values bug

zitorelova · zitorelova · commit 4f0d461b6c83 · 2021-01-28T19:46:59.000-08:00
diff --git a/pandas/core/frame.py b/pandas/core/frame.py
@@ -5554,7 +5554,9 @@ def sort_values(  # type: ignore[override]
         )
 
         if ignore_index:
-            new_data.set_axis(1, ibase.default_index(len(indexer)))
+            new_data.set_axis(
+                self._get_block_manager_axis(axis),
+                ibase.default_index(len(indexer)))
 
         result = self._constructor(new_data)
         if inplace:

Original file line number	Diff line number	Diff line change
`@@ -5554,7 +5554,9 @@ def sort_values( # type: ignore[override]`
`5554`	`5554`	`)`
`5555`	`5555`
`5556`	`5556`	`if ignore_index:`
`5557`		`- new_data.set_axis(1, ibase.default_index(len(indexer)))`
	`5557`	`+ new_data.set_axis(`
	`5558`	`+ self._get_block_manager_axis(axis),`
	`5559`	`+ ibase.default_index(len(indexer)))`
`5558`	`5560`
`5559`	`5561`	`result = self._constructor(new_data)`
`5560`	`5562`	`if inplace:`