《Python数据科学指南》——1.15　使用装饰器改变函数行为

本节书摘来自异步社区《Python数据科学指南》一书中的第1章，第1.15节，作者[印度] Gopi Subramanian ，方延风 刘丹 译，更多章节内容可以访问云栖社区“异步社区”公众号查看。

1.15 使用装饰器改变函数行为

装饰器能封装一个函数，并改变它的行为，通过示例是理解它们的最好方式，本节中我们演示了实际应用中的一些示例。

1.15.1 准备工作

还记得我们在前面章节中将函数作为另一个函数的参数、函数作为一个变量、函数中返回函数等介绍吗？最重要的是，你还记得那个圆柱体的例子吗？如果你掌握了这些，装饰器只是小菜一碟。在本节的示例中，我们将对给定的字符串建立清理操作管道：给定一个混合大小写并带有标点符号的字符串，我们使用装饰器对它进行清理，这些操作还很容易进行扩展。

1.15.2 操作方法

我们写一个简单的装饰器来进行文本操作。

from string import punctuation

def pipeline_wrapper(func):

 def to_lower(x):

 return x.lower()

 def remove_punc(x):

 for p in punctuation:

 x = x.replace(p,)

 return x

 def wrapper(*args,**kwargs):

 x = to_lower(*args,**kwargs)

 x = remove_punc(x)

 return func(x)

 return wrapper

@pipeline_wrapper

def tokenize_whitespace(inText):

 return inText.split()

s = "string. With. Punctuation?"

print tokenize_whitespace(s)

1.15.3 工作原理

我们先从以下两行开始。

s = "string. With. Punctuation?"

print tokenize_whitespace(s)

我们声明了一个字符串变量，然后想对它进行清理，使之满足以下特性。

将字符串转为小写。

清除标点符号。

返回一个词列表。

我们用字符串s作为参数，调用了tokenize_whitespace函数，我们来看看这个函数。

@pipeline_wrapper

def tokenize_whitespace(inText):

return inText.split()

这个函数很简单：输入一个字符串，函数采用空格作为分隔符将它进行分割，并返回一个词列表。接下来我们使用装饰器来改变这个函数的行为，这个装饰器就是@pipeline_wrapper，它是以简便的方式调用以下语句。

tokenize_whitespace = pipeline_wrapper (clean_tokens)

我们仔细看看这个装饰器函数。

def pipeline_wrapper(func):

 def to_lower(x):

 return x.lower()

 def remove_punc(x):

 for p in punctuation:

 x = x.replace(p,)

 return x

 def wrapper(*args,**kwargs):

 x = to_lower(*args,**kwargs)

 x = remove_punc(x)

 return func(x)

 return wrapper

pipeline_wrapper返回了wrapper函数，在后者中，最后的返回语句是返回func，这是我们传递给wrapper的原始函数，wrapper改变了我们原来的pipeline_wrapper函数的行为。pipeline_wrapper的输入先被to_lower()函数修改了，转成了小写。随后是remove_punc()函数，将标点符号清除。最后的输出如下。

[string, with, punctuation]

以上结果就是我们所要的：清除标点符号，转为小写字符，最后形式是词的列表。

异步社区 异步社区(www.epubit.com)是人民邮电出版社旗下IT专业图书旗舰社区，也是国内领先的IT专业图书社区，致力于优质学习内容的出版和分享，实现了纸书电子书的同步上架，于2015年8月上线运营。公众号【异步图书】，每日赠送异步新书。