Skip to content

Commit 01f792b

Browse files
author
tp
committed
Updated according to comments (GH 17414)
1 parent 630b664 commit 01f792b

File tree

3 files changed

+12
-11
lines changed

3 files changed

+12
-11
lines changed

pandas/core/indexes/base.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -2533,7 +2533,9 @@ def _get_level_values(self, level):
25332533
25342534
Parameters
25352535
----------
2536-
level : int or level name
2536+
level : int or str
2537+
``level`` is either the integer position of the level in the
2538+
MultiIndex, or the name of the level.
25372539
25382540
Returns
25392541
-------

pandas/core/indexes/multi.py

+8-9
Original file line numberDiff line numberDiff line change
@@ -886,7 +886,9 @@ def get_level_values(self, level):
886886
887887
Parameters
888888
----------
889-
level : int or level name
889+
level : int or str
890+
``level`` is either the integer position of the level in the
891+
MultiIndex, or the name of the level.
890892
891893
Returns
892894
-------
@@ -897,18 +899,15 @@ def get_level_values(self, level):
897899
898900
Create a MultiIndex:
899901
900-
>>> i1 = pd.Index(list('abc'), name='level_1')
901-
>>> i2 = pd.CategoricalIndex(list('def'), name='level_2')
902-
>>> mi = pd.MultiIndex.from_arrays((i1, i2))
902+
>>> mi = pd.MultiIndex.from_arrays((list('abc'), list('def')))
903+
>>> mi.names = ['level_1', 'level_2']
903904
904905
Get level values by supplying level as either integer or name:
905906
906-
>>> mi.get_level_values(1)
907-
CategoricalIndex(['d', 'e', 'f'], categories=['d', 'e', 'f'],
908-
ordered=False, name='level_2',
909-
dtype='category')
910-
>>> mi.get_level_values('level_1')
907+
>>> mi.get_level_values(0)
911908
Index(['a', 'b', 'c'], dtype='object', name='level_1')
909+
>>> mi.get_level_values('level_2')
910+
Index(['d', 'e', 'f'], dtype='object', name='level_2')
912911
"""
913912
level = self._get_level_number(level)
914913
values = self._get_level_values(level)

pandas/tests/indexes/test_base.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -1438,7 +1438,7 @@ def test_get_level_values(self):
14381438
result = self.strIndex.get_level_values(0)
14391439
tm.assert_index_equal(result, self.strIndex)
14401440

1441-
# test with name
1441+
# GH 17414
14421442
index_with_name = self.strIndex.copy()
14431443
index_with_name.name = 'a'
14441444
result = index_with_name.get_level_values('a')

0 commit comments

Comments
 (0)