Pipe——Python 的中綴文法庫

賴勇浩（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 linux2Type "help", "copyright", "credits" or "license" for more information.>>> from pipe import *>>> [1, 2, 3, 4, 5] | add15>>> [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) | add34

因為 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 更奇幻的模組，希望我能理解清楚，能夠跟大家介紹之。