1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
| import numpy as np
| import pytest
|
| from pandas import (
| DataFrame,
| MultiIndex,
| )
| import pandas._testing as tm
|
|
| class TestReorderLevels:
| def test_reorder_levels(self, frame_or_series):
| index = MultiIndex(
| levels=[["bar"], ["one", "two", "three"], [0, 1]],
| codes=[[0, 0, 0, 0, 0, 0], [0, 1, 2, 0, 1, 2], [0, 1, 0, 1, 0, 1]],
| names=["L0", "L1", "L2"],
| )
| df = DataFrame({"A": np.arange(6), "B": np.arange(6)}, index=index)
| obj = tm.get_obj(df, frame_or_series)
|
| # no change, position
| result = obj.reorder_levels([0, 1, 2])
| tm.assert_equal(obj, result)
|
| # no change, labels
| result = obj.reorder_levels(["L0", "L1", "L2"])
| tm.assert_equal(obj, result)
|
| # rotate, position
| result = obj.reorder_levels([1, 2, 0])
| e_idx = MultiIndex(
| levels=[["one", "two", "three"], [0, 1], ["bar"]],
| codes=[[0, 1, 2, 0, 1, 2], [0, 1, 0, 1, 0, 1], [0, 0, 0, 0, 0, 0]],
| names=["L1", "L2", "L0"],
| )
| expected = DataFrame({"A": np.arange(6), "B": np.arange(6)}, index=e_idx)
| expected = tm.get_obj(expected, frame_or_series)
| tm.assert_equal(result, expected)
|
| result = obj.reorder_levels([0, 0, 0])
| e_idx = MultiIndex(
| levels=[["bar"], ["bar"], ["bar"]],
| codes=[[0, 0, 0, 0, 0, 0], [0, 0, 0, 0, 0, 0], [0, 0, 0, 0, 0, 0]],
| names=["L0", "L0", "L0"],
| )
| expected = DataFrame({"A": np.arange(6), "B": np.arange(6)}, index=e_idx)
| expected = tm.get_obj(expected, frame_or_series)
| tm.assert_equal(result, expected)
|
| result = obj.reorder_levels(["L0", "L0", "L0"])
| tm.assert_equal(result, expected)
|
| def test_reorder_levels_swaplevel_equivalence(
| self, multiindex_year_month_day_dataframe_random_data
| ):
| ymd = multiindex_year_month_day_dataframe_random_data
|
| result = ymd.reorder_levels(["month", "day", "year"])
| expected = ymd.swaplevel(0, 1).swaplevel(1, 2)
| tm.assert_frame_equal(result, expected)
|
| result = ymd["A"].reorder_levels(["month", "day", "year"])
| expected = ymd["A"].swaplevel(0, 1).swaplevel(1, 2)
| tm.assert_series_equal(result, expected)
|
| result = ymd.T.reorder_levels(["month", "day", "year"], axis=1)
| expected = ymd.T.swaplevel(0, 1, axis=1).swaplevel(1, 2, axis=1)
| tm.assert_frame_equal(result, expected)
|
| with pytest.raises(TypeError, match="hierarchical axis"):
| ymd.reorder_levels([1, 2], axis=1)
|
| with pytest.raises(IndexError, match="Too many levels"):
| ymd.index.reorder_levels([1, 2, 3])
|
|
