Pipe——Python 的中缀语法库

petrel919

　　赖勇浩（http://laiyonghao.com）
　　注：本文基本上是翻译这篇文章（http://dev-tricks.net/pipe-infix-syntax-for-python）。
　　通过 Pipe 模块，就能够使用 Python 用上中缀语法。
　　首先来看一下传统前缀语法的代码：

sum(select(where(take_while(fib(), lambda x: x < 1000000) lambda x: x % 2), lambda x: x * x))
　　很难读？再来看看中缀语法代码：

fib() | take_while(lambda x: x < 1000000) \
| where(lambda x: x % 2) \
| select(lambda x: x * x) \
| sum()
　　好读多了吧？
　　虽然 Pipe 基类的代码很少，但很强大，能够让你很容易写出 pipeable 函数哦。而且这个模块本身就带了超过 30 个已经写好的函数，比如 ‘where’, ‘group_by’, ‘sort’, ‘take_while’ …
　　如果想一下 Pipe，需要先安装，在命令行执行：

pip install -U pipe
　　然后等着安装完成就行了。现在可以打开一个交互式 Shell，来试一下：

Python 2.6.6 (r266:84292, Dec 26 2010, 22:31:48)
[GCC 4.4.5] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>> from pipe import *
>>> [1, 2, 3, 4, 5] | add
15
>>> [5, 4, 3, 2, 1] | sort
[1, 2, 3, 4, 5]
　　很简单吧？如果有什么问题，可以随时 help(pipe) 一下，就可以看到完备的帮助了。
　　接下来我们展示一下组合两个或更多的 pipeable 函数：

>>> [1, 2, 3, 4, 5] | where(lambda x: x % 2) | concat
'1, 3, 5'
>>> [1, 2, 3, 4, 5] | where(lambda x: x % 2) | tail(2) | concat
'3, 5'
>>> [1, 2, 3, 4, 5] | where(lambda x: x % 2) | tail(2) | select(lambda x: x * x) | concat
'9, 25'
>>> [1, 2, 3, 4, 5] | where(lambda x: x % 2) | tail(2) | select(lambda x: x * x) | add
34
　　因为 Pipe 是惰性求值的，所以我们完成可以弄一个无穷生成器而不用担心内存用完，比如：

>>> def fib():
...    a, b = 0, 1
...    while True:
...        yield a
...        a, b = b, a + b
　　现在让我们用 fib() 函数来完成一个 http://projecteuler.net 的第 2 题：

　　Find the sum of all the even-valued terms in Fibonacci which do not exceed four million.

>>> euler2 = fib() | where(lambda x: x % 2 == 0) | take_while(lambda x: x < 4000000) | add
>>> assert euler2 == 4613732
　　怎么样？可读性强吧？漂亮不？
　　最后，我们来学习一下如何利用 @Pipe decorator 创建一个新的 pipeable 函数：
　　假定要创建一个函数 yield 它的输入的前 x 个元素
　　假定要创建一个函数能够用以 (1, 2, 3, 4, 5) | take(2) 语句来获取前两个元素
　　那么最初的实现可能是这样的：

def take(iterable, qte):
for item in iterable:
if qte > 0:
qte -= 1
yield item
else:
return
　　现在，你只要把 @Pipe 盖在这个函数上头，这货就是 pipeable 函数了！
　　====================
　　鸣谢：
　　感谢 @yinhm 在 Twitter 上分享《Pipe: Infix syntax for Python》一文，让我知道还有这等神器。
　　感谢 @kyhpudding 在 Twitter 上分享他的 solo 模块，一个比 pipe 更奇幻的模块，希望我能理解清楚，能够跟大家介绍之。