Skip to content

Commit

Permalink
Deprecate .add_prefix and .add_suffix
Browse files Browse the repository at this point in the history
  • Loading branch information
tp committed Dec 24, 2017
1 parent cdebcf3 commit 3a5785a
Show file tree
Hide file tree
Showing 3 changed files with 54 additions and 29 deletions.
16 changes: 7 additions & 9 deletions doc/source/10min.rst
Original file line number Diff line number Diff line change
Expand Up @@ -87,15 +87,13 @@ will be completed:
df2.A df2.bool
df2.abs df2.boxplot
df2.add df2.C
df2.add_prefix df2.clip
df2.add_suffix df2.clip_lower
df2.align df2.clip_upper
df2.all df2.columns
df2.any df2.combine
df2.append df2.combine_first
df2.apply df2.compound
df2.applymap df2.consolidate
df2.D
df2.align df2.clip
df2.all df2.clip_lower
df2.any df2.clip_upper
df2.append df2.columns
df2.apply df2.combine
df2.applymap df2.combine_first
df2.B df2.compound

As you can see, the columns ``A``, ``B``, ``C``, and ``D`` are automatically
tab completed. ``E`` is there as well; the rest of the attributes have been
Expand Down
39 changes: 31 additions & 8 deletions pandas/core/generic.py
Original file line number Diff line number Diff line change
Expand Up @@ -114,8 +114,9 @@ class NDFrame(PandasObject, SelectionMixin):
'__array_interface__']
_internal_names_set = set(_internal_names)
_accessors = frozenset([])
_deprecations = frozenset(['as_blocks', 'blocks',
'consolidate', 'convert_objects', 'is_copy'])
_deprecations = frozenset(['as_blocks', 'add_prefix', 'add_suffix',
'blocks', 'consolidate', 'convert_objects',
'is_copy'])
_metadata = []
_is_copy = None

Expand Down Expand Up @@ -801,6 +802,11 @@ def swaplevel(self, i=-2, j=-1, axis=0):
1 2
4 3
dtype: int64
>>> s.rename('index_{}'.format) # function, changes labels
index_0 1
index_1 2
index_2 3
dtype: object
>>> s.rename({1: 3, 2: 5}) # mapping, changes labels
0 1
3 2
Expand All @@ -824,11 +830,11 @@ def swaplevel(self, i=-2, j=-1, axis=0):
We *highly* recommend using keyword arguments to clarify your
intent.
>>> df.rename(index=str, columns={"A": "a", "B": "c"})
a c
0 1 4
1 2 5
2 3 6
>>> df.rename(index="index_{}".format, columns={"A": "a", "B": "c"})
a c
index_0 1 4
index_1 2 5
index_2 3 6
>>> df.rename(index=str, columns={"A": "a", "C": "c"})
a B
Expand Down Expand Up @@ -2922,6 +2928,9 @@ def _update_inplace(self, result, verify_is_copy=True):

def add_prefix(self, prefix):
"""
DEPRECATED: Use ``obj.rename(columns=lambda x: 'prefix_'+str(x))`` or
similar instead.
Concatenate prefix string with panel items names.
Parameters
Expand All @@ -2931,13 +2940,21 @@ def add_prefix(self, prefix):
Returns
-------
with_prefix : type of caller
See Also:
---------
rename : Alter axes labels.
"""
warnings.warn("'.add_prefix' is deprecated and will be removed in a "
"future version. Use '.rename' instead",
FutureWarning, stacklevel=2)
new_data = self._data.add_prefix(prefix)
return self._constructor(new_data).__finalize__(self)

def add_suffix(self, suffix):
"""
Concatenate suffix string with panel items names.
DEPRECATED: Use ``obj.rename(columns=lambda x: 'prefix_'+str(x))`` or
similar instead. Concatenate suffix string with panel items names.
Parameters
----------
Expand All @@ -2946,7 +2963,13 @@ def add_suffix(self, suffix):
Returns
-------
with_suffix : type of caller
See Also:
---------
rename : Alter axes labels.
"""
warnings.warn("'add_suffix' is deprecated and will be removed in a "
"future version.", FutureWarning, stacklevel=2)
new_data = self._data.add_suffix(suffix)
return self._constructor(new_data).__finalize__(self)

Expand Down
28 changes: 16 additions & 12 deletions pandas/tests/frame/test_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -76,21 +76,25 @@ def test_get_value(self):
tm.assert_almost_equal(result, expected)

def test_add_prefix_suffix(self):
with_prefix = self.frame.add_prefix('foo#')
expected = pd.Index(['foo#%s' % c for c in self.frame.columns])
tm.assert_index_equal(with_prefix.columns, expected)
with tm.assert_produces_warning(FutureWarning):
with_prefix = self.frame.add_prefix('foo#')
expected = pd.Index(['foo#%s' % c for c in self.frame.columns])
tm.assert_index_equal(with_prefix.columns, expected)

with_suffix = self.frame.add_suffix('#foo')
expected = pd.Index(['%s#foo' % c for c in self.frame.columns])
tm.assert_index_equal(with_suffix.columns, expected)
with tm.assert_produces_warning(FutureWarning):
with_suffix = self.frame.add_suffix('#foo')
expected = pd.Index(['%s#foo' % c for c in self.frame.columns])
tm.assert_index_equal(with_suffix.columns, expected)

with_pct_prefix = self.frame.add_prefix('%')
expected = pd.Index(['%{}'.format(c) for c in self.frame.columns])
tm.assert_index_equal(with_pct_prefix.columns, expected)
with tm.assert_produces_warning(FutureWarning):
with_pct_prefix = self.frame.add_prefix('%')
expected = pd.Index(['%{}'.format(c) for c in self.frame.columns])
tm.assert_index_equal(with_pct_prefix.columns, expected)

with_pct_suffix = self.frame.add_suffix('%')
expected = pd.Index(['{}%'.format(c) for c in self.frame.columns])
tm.assert_index_equal(with_pct_suffix.columns, expected)
with tm.assert_produces_warning(FutureWarning):
with_pct_suffix = self.frame.add_suffix('%')
expected = pd.Index(['{}%'.format(c) for c in self.frame.columns])
tm.assert_index_equal(with_pct_suffix.columns, expected)

def test_get_axis(self):
f = self.frame
Expand Down

0 comments on commit 3a5785a

Please sign in to comment.