Skip to content

Commit 37ceb5d

Browse files
committed
1 parent 0c4e611 commit 37ceb5d

File tree

2 files changed

+7
-5
lines changed

2 files changed

+7
-5
lines changed

pandas/core/generic.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -933,6 +933,7 @@ def rename(self, *args, **kwargs):
933933
inplace = kwargs.pop('inplace', False)
934934
level = kwargs.pop('level', None)
935935
axis = kwargs.pop('axis', None)
936+
tupleize_cols = kwargs.pop('tupleize_cols', False)
936937
if axis is not None:
937938
axis = self._get_axis_number(axis)
938939

@@ -971,7 +972,7 @@ def f(x):
971972
if level is not None:
972973
level = self.axes[axis]._get_level_number(level)
973974
result._data = result._data.rename_axis(f, axis=baxis, copy=copy,
974-
level=level)
975+
level=level, tupleize_cols=tupleize_cols)
975976
result._clear_item_cache()
976977

977978
if inplace:

pandas/core/internals.py

+5-4
Original file line numberDiff line numberDiff line change
@@ -3280,7 +3280,8 @@ def set_axis(self, axis, new_labels):
32803280

32813281
self.axes[axis] = new_labels
32823282

3283-
def rename_axis(self, mapper, axis, copy=True, level=None):
3283+
def rename_axis(self, mapper, axis, copy=True, level=None,
3284+
tupleize_cols=False):
32843285
"""
32853286
Rename one of axes.
32863287
@@ -3293,7 +3294,7 @@ def rename_axis(self, mapper, axis, copy=True, level=None):
32933294
32943295
"""
32953296
obj = self.copy(deep=copy)
3296-
obj.set_axis(axis, _transform_index(self.axes[axis], mapper, level))
3297+
obj.set_axis(axis, _transform_index(self.axes[axis], mapper, level, tupleize_cols=tupleize_cols))
32973298
return obj
32983299

32993300
def add_prefix(self, prefix):
@@ -5234,7 +5235,7 @@ def _safe_reshape(arr, new_shape):
52345235
return arr
52355236

52365237

5237-
def _transform_index(index, func, level=None):
5238+
def _transform_index(index, func, level=None, tupleize_cols=True):
52385239
"""
52395240
Apply function to all values found in index.
52405241
@@ -5251,7 +5252,7 @@ def _transform_index(index, func, level=None):
52515252
return MultiIndex.from_tuples(items, names=index.names)
52525253
else:
52535254
items = [func(x) for x in index]
5254-
return Index(items, name=index.name)
5255+
return Index(items, name=index.name, tupleize_cols=tupleize_cols)
52555256

52565257

52575258
def _putmask_smart(v, m, n):

0 commit comments

Comments
 (0)