《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']

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

相关资源：敏捷开发V1.0.pptx