之前已发布
DataFrame的apply和applymap方法以及Series中的map方法
利用Pandas进行数据转换(map、replace、rename、duplicated等函数的使用)
本文再补充更多的关于replace函数的用法,以及没有提到的clip函数。
clip函数
准备示例数据:
>>> data = {'col_0': [9, -3, 0, -1, 5], 'col_1': [-2, -7, 6, 8, -5]}
>>> df = pd.DataFrame(data)
>>> df
col_0 col_1
0 9 -2
1 -3 -7
2 0 6
3 -1 8
4 5 -5
Clips对超过或者低于指定阈值的数据将替换为指定阈值:
>>> df.clip(-4, 6)
col_0 col_1
0 6 -2
1 -3 -4
2 0 6
3 -1 6
4 5 -4
传入两个Series,并指定axis=0则可以对每一行数据都用指定的范围进行截断:
>>> df
col_0 col_1
0 9 -2
1 -3 -7
2 0 6
3 -1 8
4 5 -5
>>> t1 = pd.Series([2, -4, -1, 6, 3])
>>> t2 = pd.Series([6, 0, 3, 9, 8])
>>> df.clip(t1, t2, axis=0)
col_0 col_1
0 6 2
1 -3 -4
2 0 3
3 6 8
4 5 3
比如对于第一行数据9 和 -2被截断在[2,6]的范围,大于6取6,小于2取2.
DataFrame或Series的replace方法
对指定的单个标量值进行替换:
>>> s = pd.Series([0, 1, 2, 3, 4])
>>> s.replace(0, 5)
0 5
1 1
2 2
3 3
4 4
dtype: int64
>>> df = pd.DataFrame({'A': [0, 1, 2, 3, 4],
... 'B': [5, 6, 7, 8, 9],
... 'C': ['a', 'b', 'c', 'd', 'e']})
>>> df.replace(0, 5)
A B C
0 5 5 a
1 1 6 b
2 2 7 c
3 3 8 d
4 4 9 e
将指定的列表内的所有标量值都替换为指定值:
>>> df.replace([0, 1, 2, 3], 4)
A B C
0 4 5 a
1 4 6 b
2 4 7 c
3 4 8 d
4 4 9 e
传入两个长度一致的列表,则有一一对应的替换关系:
>>> df.replace([0, 1, 2, 3], [4, 3, 2, 1])
A B C
0 4 5 a
1 3 6 b
2 2 7 c
3 1 8 d
4 4 9 e
其中0替换为4,1替换为3,2依然替换为2,替换为1。
对指定标量值使用向上填充的方式进行替换:
>>> s.replace([1, 2], method='bfill')
0 0
1 3
2 3
3 3
4 4
dtype: int64
原本的1和2都被下面3向上填充。
使用单个普通字典进行替换,效果与传入两个长度一致的列表一致:
>>> df.replace({0: 10, 1: 100})
A B C
0 10 5 a
1 100 6 b
2 2 7 c
3 3 8 d
4 4 9 e
传入两个参数,则字典中的键为要被替换的键,值为要被替换的值,第二个参数为替换后的值:
>>> df.replace({'A': 0, 'B': 5}, 100)
A B C
0 100 100 a
1 1 6 b
2 2 7 c
3 3 8 d
4 4 9 e
还可以传入嵌套字典,外层字典的键表示要替换的列,内层字典的键值表示被替换的值和替换后的值:
>>> df.replace({'A': {0: 100, 4: 400}})
A B C
0 100 5 a
1 1 6 b
2 2 7 c
3 3 8 d
4 400 9 e
指定参数regex=True即可实现正则表达式替换。
>>> df = pd.DataFrame({'A': ['bat', 'foo', 'bait'],
... 'B': ['abc', 'bar', 'xyz']})
>>> df.replace(to_replace=r'^ba.$', value='new', regex=True)
A B
0 new abc
1 foo new
2 bait xyz
下面将A列符合^ba.$规则的都替换为new:
>>> df.replace({'A': r'^ba.$'}, {'A': 'new'}, regex=True)
A B
0 new abc
1 foo bar
2 bait xyz
也可以直接向regex参数传递替换规则:
>>> df.replace(regex=r'^ba.$', value='new')
A B
0 new abc
1 foo new
2 bait xyz
>>> df.replace(regex={r'^ba.$': 'new', 'foo': 'xyz'})
A B
0 new abc
1 xyz new
2 bait xyz
>>> df.replace(regex=[r'^ba.$', 'foo'], value='new')
A B
0 new abc
1 new new
2 bait xyz
Tips:传入一个字典和传入两个替换参数的默认行为不一样。
s = pd.Series([10, 'a', 'a', 'b', 'a'])
s.replace({'a': None})
相当于:
s.replace(to_replace={'a': None}, value=None, method=None)
>>> s.replace({'a': None})
0 10
1 None
2 None
3 b
4 None
dtype: object
s.replace('a', None)
相当于:
s.replace(to_replace='a', value=None, method='pad')
>>> s.replace('a', None)
0 10
1 10
2 10
3 b
4 b
dtype: object
Series.str.replace
Series.str.replace默认支持正则表达式:
>>> se1 = pd.Series(['foo', 'fuz', np.nan])
>>> se1.str.replace('f.', 'ba')
0 bao
1 baz
2 NaN
dtype: object
也可以设置参数regex=False关闭:
>>> se1.str.replace('f.', 'ba', regex=False)
0 bao
1 fuz
2 NaN
dtype: object
当给第二个参数repl传递一个函数时,该函数接收一个正则表达式对象,返回一个字符串作为替换结果:
>>> def f(arg):
print(arg, type(arg))
return arg.group(0)
>>> se1.str.replace('f.', f)
0 foo
1 fuz
2 NaN
dtype: object
根据此特性实现将每个小写单词都倒序:
>>> repl = lambda m: m.group(0)[::-1]
>>> se2 = pd.Series(['foo 123', 'bar baz', np.nan])
>>> se2.str.replace(r'[a-z]+', repl)
0 oof 123
1 rab zab
2 NaN
dtype: object
使用正则表达式的组(提取第二组并交换大小写):
pat = r"(?P\w+) (?P\w+) (?P\w+)"
repl = lambda m: m.group('two').swapcase()
pd.Series(['One Two Three', 'Foo Bar Baz']).str.replace(pat, repl)
0 tWO
1 bAR
dtype: object
还可以使用python编译好的正则表达式对象便于复用:
import re
regex_pat = re.compile(r'FUZ', flags=re.IGNORECASE)
pd.Series(['foo', 'fuz', np.nan]).str.replace(regex_pat, 'bar')
0 foo
1 bar
2 NaN
dtype: object