寫 Python 代碼不可不知的函數式編程技術

選自 Medium

作者：Raivat Shah

參與：魔王、Jamin

本文對 Python 中的函數式編程技術進行了簡單的入門介紹。

近來，越來越多人使用函數式編程（functional programming）。因此，很多傳統的命令式語言（如 Java 和 Python）開始支持函數式編程技術。本文對 Python 中的函數式編程技術進行了簡單的入門介紹。

本文適合對函數式編程有基本了解的讀者。如果你對函數式編程並不熟悉，可以先閱讀這篇文章：https://medium.com/@yannickdot/functional-programming-101-6bc132674ec5。

本文作者是新加坡國立大學計算機學院和「USP」博學計劃學生 Raivat Shah，專注於編程與數據研究。

頭等函數

在 Python 中，函數是「頭等公民」（first-class）。也就是說，函數與其他數據類型（如 int）處於平等地位。

因而，我們可以將函數賦值給變量，也可以將其作為參數傳入其他函數，將它們存儲在其他數據結構（如 dicts）中，並將它們作為其他函數的返回值。

把函數作為對象

由於其他數據類型（如 string、list 和 int）都是對象，那麽函數也是 Python 中的對象。我們來看示例函數 foo，它將自己的名稱列印出來：

由於函數是對象，因此我們可以將函數 foo 賦值給任意變量，然後調用該變量。例如，我們可以將函數賦值給變量 bar：

語句 bar = foo 將函數 foo 引用的對象賦值給變量 bar。

把對象作為函數

當對象可調用時（callable），它們與函數一樣，如 object()。這是通過 __call__ 方法實現的。

示例如下：

每一次配置 Greeter 類的對象時，我們都會創建一個新的對象，即打招呼時可以喊的新名字。如下所示：

我們可以調用 morning 對象的原因在於，我們已經在類定義中使用了 __call__ 方法。為了檢查對象是否可調用，我們使用內置函數 callable：

數據結構內的函數

函數和其他對象一樣，可以存儲在數據結構內部。例如，我們可以創建 int to func 的字典。當 int 是待執行步驟的簡寫時，這就會派上用場。

類似地，函數也可以存儲在多種其他數據結構中。

把函數作為參數和返回值

函數還可以作為其他函數的參數和返回值。接受函數作為輸入或返回函數的函數叫做高階函數，它是函數式編程的重要組成部分。

高階函數具備強大的能力。就像《Eloquent JavaScript》中解釋的那樣：

「高階函數允許我們對動作執行抽象，而不只是抽象數值。」

我們來看一個例子。假設我們想對一個項目列表（list of items）執行迭代，並將其順序列印出來。我們可以輕鬆構建一個 iterate 函數：

看起來很酷吧，但這只不過是一級抽象而已。如果我們想在對列表執行迭代時進行列印以外的其他操作要怎麽做呢？

這就是高階函數存在的意義。我們可以創建函數 iterate_custom，待執行迭代的列表和要對每個項應用的函數都是 iterate_custom 函數的輸入：

這看起來微不足道，但其實非常強大。

我們已經把抽象的級別提高了一層，使代碼具備更強的可重用性。現在，我們不僅可以在列印列表時調用該函數，還可以對涉及序列迭代的列表執行任意操作。

函數還能被返回，從而使事情變得更加簡單。就像我們在 dict 中存儲函數一樣，我們還可以將函數作為控制語句，來決定適合的函數。例如：

嵌套函數

函數還可以在其他函數內部，這就是「內部函數」。內部函數在創建輔助函數時非常有用，輔助函數即作為子模塊來支持主函數的小型可重用函數。

在問題需要特定函數定義（參數類型或順序）時，我們可以使用輔助函數。這種不遵循傳統做法的操作使得解決問題變得更加簡單，示例參見：http://www-inst.eecs.berkeley.edu/~cs61a/sp12/lectures/lect4-2x3.pdf。

假設你想定義一個斐波那契函數 fib(n)，該函數只有一個參數 n，我們必須返回第 n 個斐波那契數。

定義此類函數的一種可行方式是：使用輔助函數來追蹤斐波那契數列的前兩個項（因為斐波那契數是前兩個數之和）。

將該計算從函數主體移到函數參數，這具備非常強大的力量。因為它減少了遞歸方法中可能出現的冗余計算。

單表達式函數（Lambda 表達式）

如果我們想在未給函數命名之前寫一個函數要怎麽做？如果我們想寫一個簡短的單行函數（如上述示例中的函數 foo 或 mult）要怎麽做？

我們可以在 Python 中使用 lambda 關鍵字來定義此類函數。示例如下：

該 mult 函數的行為與使用傳統 def 關鍵字定義函數的行為相同。

注意：lambda 函數必須為單行，且不能包含程序員寫的返回語句。

事實上，它們通常具備隱式的返回語句（在上面的示例中，函數想表達 return x * y，不過我們省略了 lambda 函數中的顯式返回語句）。

lambda 函數更加強大和精準，因為我們還可以構建匿名函數（即沒有名稱的函數）：

當我們只需要一次性使用某函數時，這種方法非常方便。例如，當我們想填充字典時：

接下來我們來看 Map、Filter 和 Reduce，以更多地了解 lambda。

Map、Filter 和 Reduce

Map

map 函數基於指定過程（函數）將輸入集轉換為另一個集合。這類似於上文提到的 iterate_custom 函數。例如：

在 Python 3 中，map 函數返回的 map 對象可被類型轉換為 list，以方便使用。現在，我們無需顯式地定義 multiply_by_four 函數，而是定義 lambda 表達式：

當我們想對集合內的所有值執行某項操作時，map 函數很有用。

Filter

就像名稱所顯示的那樣，filter 函數可以幫助篩除不想要的項。例如，我們想要去除 scores 中的奇數，那麽我們可以使用 filter：

由於提供給 filter 的函數是逐個決定是否接受每一個項的，因此該函數必須返回 bool 值，且該函數必須是一元函數（即隻使用一個輸入參數）。

Reduce

reduce 函數用於「總結」或「概述」數據集。例如，如果我們想要計算所有分數的總和，就可以使用 reduce：

這要比寫循環語句簡單多了。注意：提供給 reduce 的函數需要兩個參數：一個表示正在接受檢查的項，另一個表示所用運算的累積結果。

本文是關於函數式編程的一篇入門文章，雖然盡量完備地介紹了相關的知識，但並不是那麽深入。如想了解更多，大家可以閱讀以下資源：

Best Practices for Using Functional Programming in Python：https://kite.com/blog/python/functional-programming/

Functional Programming Tutorials and Notes：https://www.hackerearth.com/zh/practice/python/functional-programming/functional-programming-1/tutorial/

本文為機器之心編譯，轉載請聯繫本公眾號獲得授權。

------------------------------------------------