We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0f30bea commit 6be7e1aCopy full SHA for 6be7e1a
pandas/tests/indexes/multi/test_indexing.py
@@ -455,3 +455,19 @@ def test_multiindex_loc_order():
455
res = df.loc[(["b", "a"], [2, 1]), :]
456
exp_index = pd.MultiIndex.from_arrays([["b", "b", "a", "a"], [2, 1, 2, 1]])
457
tm.assert_index_equal(res.index, exp_index)
458
+
459
+ res = df.loc[:, ["Colorado", "Ohio"]]
460
+ exp_columns = pd.MultiIndex.from_arrays(
461
+ [["Colorado", "Ohio", "Ohio"], ["Green", "Green", "Red"]]
462
+ )
463
+ tm.assert_index_equal(res.columns, exp_columns)
464
465
+ res = df.loc[:, (["Colorado", "Ohio"], ["Red", "Green"])]
466
467
+ [["Colorado", "Ohio", "Ohio"], ["Green", "Red", "Green"]]
468
469
470
471
+ res = df.loc[(slice(None), [2, 1]), :]
472
+ exp_index = pd.MultiIndex.from_arrays([["a", "b", "a", "b"], [2, 2, 1, 1]])
473
+ tm.assert_index_equal(res.index, exp_index)
0 commit comments