@@ -2104,7 +2104,7 @@ def test_ix_general(self):
2104
2104
tm .assert_frame_equal (df .ix [key ], df .iloc [2 :])
2105
2105
2106
2106
# this is ok
2107
- df .sortlevel (inplace = True )
2107
+ df .sort_index (inplace = True )
2108
2108
res = df .ix [key ]
2109
2109
# col has float dtype, result should be Float64Index
2110
2110
index = MultiIndex .from_arrays ([[4. ] * 3 , [2012 ] * 3 ],
@@ -2137,7 +2137,7 @@ def test_xs_multiindex(self):
2137
2137
[('a' , 'foo' ), ('a' , 'bar' ), ('b' , 'hello' ),
2138
2138
('b' , 'world' )], names = ['lvl0' , 'lvl1' ])
2139
2139
df = DataFrame (np .random .randn (4 , 4 ), columns = columns )
2140
- df .sortlevel (axis = 1 , inplace = True )
2140
+ df .sort_index (axis = 1 , inplace = True )
2141
2141
result = df .xs ('a' , level = 'lvl0' , axis = 1 )
2142
2142
expected = df .iloc [:, 0 :2 ].loc [:, 'a' ]
2143
2143
tm .assert_frame_equal (result , expected )
@@ -2180,7 +2180,7 @@ def test_per_axis_per_level_getitem(self):
2180
2180
df = DataFrame (
2181
2181
np .arange (16 , dtype = 'int64' ).reshape (
2182
2182
4 , 4 ), index = index , columns = columns )
2183
- df = df .sortlevel (axis = 0 ).sortlevel (axis = 1 )
2183
+ df = df .sort_index (axis = 0 ).sort_index (axis = 1 )
2184
2184
2185
2185
# identity
2186
2186
result = df .loc [(slice (None ), slice (None )), :]
@@ -2249,7 +2249,7 @@ def f():
2249
2249
2250
2250
# not lexsorted
2251
2251
self .assertEqual (df .index .lexsort_depth , 2 )
2252
- df = df .sortlevel (level = 1 , axis = 0 )
2252
+ df = df .sort_index (level = 1 , axis = 0 )
2253
2253
self .assertEqual (df .index .lexsort_depth , 0 )
2254
2254
with tm .assertRaisesRegexp (
2255
2255
UnsortedIndexError ,
@@ -2265,11 +2265,11 @@ def test_multiindex_slicers_non_unique(self):
2265
2265
B = ['a' , 'a' , 'a' , 'a' ],
2266
2266
C = [1 , 2 , 1 , 3 ],
2267
2267
D = [1 , 2 , 3 , 4 ]))
2268
- .set_index (['A' , 'B' , 'C' ]).sortlevel ())
2268
+ .set_index (['A' , 'B' , 'C' ]).sort_index ())
2269
2269
self .assertFalse (df .index .is_unique )
2270
2270
expected = (DataFrame (dict (A = ['foo' , 'foo' ], B = ['a' , 'a' ],
2271
2271
C = [1 , 1 ], D = [1 , 3 ]))
2272
- .set_index (['A' , 'B' , 'C' ]).sortlevel ())
2272
+ .set_index (['A' , 'B' , 'C' ]).sort_index ())
2273
2273
result = df .loc [(slice (None ), slice (None ), 1 ), :]
2274
2274
tm .assert_frame_equal (result , expected )
2275
2275
@@ -2281,11 +2281,11 @@ def test_multiindex_slicers_non_unique(self):
2281
2281
B = ['a' , 'a' , 'a' , 'a' ],
2282
2282
C = [1 , 2 , 1 , 2 ],
2283
2283
D = [1 , 2 , 3 , 4 ]))
2284
- .set_index (['A' , 'B' , 'C' ]).sortlevel ())
2284
+ .set_index (['A' , 'B' , 'C' ]).sort_index ())
2285
2285
self .assertFalse (df .index .is_unique )
2286
2286
expected = (DataFrame (dict (A = ['foo' , 'foo' ], B = ['a' , 'a' ],
2287
2287
C = [1 , 1 ], D = [1 , 3 ]))
2288
- .set_index (['A' , 'B' , 'C' ]).sortlevel ())
2288
+ .set_index (['A' , 'B' , 'C' ]).sort_index ())
2289
2289
result = df .loc [(slice (None ), slice (None ), 1 ), :]
2290
2290
self .assertFalse (result .index .is_unique )
2291
2291
tm .assert_frame_equal (result , expected )
@@ -2357,7 +2357,7 @@ def test_multiindex_slicers_edges(self):
2357
2357
2358
2358
df ['DATE' ] = pd .to_datetime (df ['DATE' ])
2359
2359
df1 = df .set_index (['A' , 'B' , 'DATE' ])
2360
- df1 = df1 .sortlevel ()
2360
+ df1 = df1 .sort_index ()
2361
2361
2362
2362
# A1 - Get all values under "A0" and "A1"
2363
2363
result = df1 .loc [(slice ('A1' )), :]
@@ -2440,7 +2440,7 @@ def f():
2440
2440
df .loc ['A1' , (slice (None ), 'foo' )]
2441
2441
2442
2442
self .assertRaises (UnsortedIndexError , f )
2443
- df = df .sortlevel (axis = 1 )
2443
+ df = df .sort_index (axis = 1 )
2444
2444
2445
2445
# slicing
2446
2446
df .loc ['A1' , (slice (None ), 'foo' )]
@@ -2459,7 +2459,7 @@ def test_loc_axis_arguments(self):
2459
2459
df = DataFrame (np .arange (len (index ) * len (columns ), dtype = 'int64' )
2460
2460
.reshape ((len (index ), len (columns ))),
2461
2461
index = index ,
2462
- columns = columns ).sortlevel ().sortlevel (axis = 1 )
2462
+ columns = columns ).sort_index ().sort_index (axis = 1 )
2463
2463
2464
2464
# axis 0
2465
2465
result = df .loc (axis = 0 )['A1' :'A3' , :, ['C1' , 'C3' ]]
@@ -2551,7 +2551,7 @@ def test_per_axis_per_level_setitem(self):
2551
2551
df_orig = DataFrame (
2552
2552
np .arange (16 , dtype = 'int64' ).reshape (
2553
2553
4 , 4 ), index = index , columns = columns )
2554
- df_orig = df_orig .sortlevel (axis = 0 ).sortlevel (axis = 1 )
2554
+ df_orig = df_orig .sort_index (axis = 0 ).sort_index (axis = 1 )
2555
2555
2556
2556
# identity
2557
2557
df = df_orig .copy ()
@@ -2764,12 +2764,12 @@ def f():
2764
2764
idx = pd .MultiIndex .from_product ([['A' , 'B' , 'C' ],
2765
2765
['foo' , 'bar' , 'baz' ]],
2766
2766
names = ['one' , 'two' ])
2767
- s = pd .Series (np .arange (9 , dtype = 'int64' ), index = idx ).sortlevel ()
2767
+ s = pd .Series (np .arange (9 , dtype = 'int64' ), index = idx ).sort_index ()
2768
2768
2769
2769
exp_idx = pd .MultiIndex .from_product ([['A' ], ['foo' , 'bar' , 'baz' ]],
2770
2770
names = ['one' , 'two' ])
2771
2771
expected = pd .Series (np .arange (3 , dtype = 'int64' ),
2772
- index = exp_idx ).sortlevel ()
2772
+ index = exp_idx ).sort_index ()
2773
2773
2774
2774
result = s .loc [['A' ]]
2775
2775
tm .assert_series_equal (result , expected )
@@ -2786,7 +2786,7 @@ def f():
2786
2786
2787
2787
idx = pd .IndexSlice
2788
2788
expected = pd .Series ([0 , 3 , 6 ], index = pd .MultiIndex .from_product (
2789
- [['A' , 'B' , 'C' ], ['foo' ]], names = ['one' , 'two' ])).sortlevel ()
2789
+ [['A' , 'B' , 'C' ], ['foo' ]], names = ['one' , 'two' ])).sort_index ()
2790
2790
2791
2791
result = s .loc [idx [:, ['foo' ]]]
2792
2792
tm .assert_series_equal (result , expected )
@@ -2799,7 +2799,7 @@ def f():
2799
2799
['alpha' , 'beta' ]))
2800
2800
df = DataFrame (
2801
2801
np .random .randn (5 , 6 ), index = range (5 ), columns = multi_index )
2802
- df = df .sortlevel ( 0 , axis = 1 )
2802
+ df = df .sort_index ( level = 0 , axis = 1 )
2803
2803
2804
2804
expected = DataFrame (index = range (5 ),
2805
2805
columns = multi_index .reindex ([])[0 ])
0 commit comments