Python函數式編程:從入門到走火入魔

shl198001 8年前發布 | 18K 次閱讀 Python 函數式編程 Python開發

很多人都在談論函數式編程(Functional Programming),只是很多人站在不同的角度看到的是完全不一樣的風景。堅持實用主義的 Python 老司機們對待 FP 的態度應該更加包容,雖然他們不相信銀彈,但冥冥中似乎能感覺到 FP 暗合了 Python 教義(The Zen of Python)的某些思想,而且既然 Python 是一門多范式編程語言,并在很大程度上支持函數式編程,那就更沒有理由拒絕它。

函數式編程源自于數學理論,它似乎也更適用于數學計算相關的場景,因此本文以一個簡單的數據處理問題為例,逐步介紹 Python 函數式編程從入門到走火入魔的過程。

問題:計算 N 位同學在某份試卷的 M 道選擇題上的得分(每道題目的分值不同)。

首先來生成一組用于計算的偽造數據:

# @file: data.py
import random  
from collections import namedtuple

Student = namedtuple('Student', ['id', 'ans'])

N_Questions = 25  
N_Students = 20

def gen_random_list(opts, n):  
    return [random.choice(opts) for i in range(n)]

# 問題答案 'ABCD' 隨機
ANS   = gen_random_list('ABCD', N_Questions)  
# 題目分值 1~5 分
SCORE = gen_random_list(range(1,6), N_Questions)

QUIZE = zip(ANS, SCORE)  
students = [  
    # 學生答案為 'ABCD*' 隨機,'*' 代表未作答
    Student(_id, gen_random_list('ABCD*', N_Questions))
    for _id in range(1, N_Students+1)
]

print(QUIZE)  
# [('A', 3), ('B', 1), ('D', 1), ...
print(students)  
# [Student(id=1, ans=['C', 'B', 'A', ...

入門

首先來看常規的面向過程編程風格,我們需要遍歷每個學生,然后遍歷每個學生對每道題目的答案并與真實答案進行比較,然后將正確答案的分數累計:

import data  
def normal(students, quize):  
    for student in students:
        sid = student.id
        score = 0
        for i in range(len(quize)):
            if quize[i][0] == student.ans[i]:
                score += quize[i][1]
        print(sid, '\t', score)

print('ID\tScore\n==================')  
normal(data.students, data.quize)  
"""
ID    Score  
==================
1      5  
2      12  
...
"""

如果你覺得上面的代碼非常直觀且合乎邏輯,那說明你已經習慣按照計算機的思維模式進行思考了。通過創建嵌套兩個 for 循環來 遍歷 所有題目答案的判斷和評分,這完全是為計算機服務的思路,雖然說 Python 中的 for 循環已經比 C 語言更進了一步,通常不需要額外的狀態變量來記錄當前循環的次數,但有時候也不得不使用狀態變量,如上例中第二個循環中比較兩個列表的元素。函數式編程的一大特點就是盡量拋棄這種明顯 循環遍歷 的做法,而是把注意集中在解決問題本身,一如在現實中我們批改試卷時,只需要將兩組答案 并列 進行比較即可:

from data import students, QUIZE

student = students[0]

# 將學生答案與正確答案合并到一起
# 然后過濾出答案一致的題目
filtered = filter(lambda x: x[0] == x[1][0], zip(student.ans, QUIZE))

print(list(filtered))  
# [('A', ('A', 3)), ('D', ('D', 1)), ...]

然后再將所有正確題目的分數累加起來,即可:

from functools import reduce

reduced = reduce(lambda x, y: x + y[1][1], filtered, 0)  
print(reduced)

以上是對一位學生的結果處理,接下來只需要對所有學生進行同樣的處理即可:

def cal(student):  
    filtered = filter(lambda x: x[0] == x[1][0], zip(student.ans, QUIZE))
    reduced = reduce(lambda x, y: x + y[1][1], filtered, 0)
    print(student.id, '\t', reduced)

print('ID\tScore\n==================')  
# 由于 Python 3 中 map 方法只是組合而不直接執行
# 需要轉換成 list 才能將 cal 方法的的結果打印出來
list(map(cal, students))

"""
ID    Score  
==================
1      5  
2      12  
...
"""

上面的示例通過 zip/filter/reduce/map 等函數將數據處理的方法 打包 應用到數據上,實現了基本的函數式編程操作。但是如果你對函數式有更深入的了解,你就會發現上面的 cal 方法中使用了全局變量 QUIZE ,這會導致在相同輸入的條件下,函數可能產生不同的輸出,這是 FP 的大忌,因此需要進行整改:

def cal(quize):  
    def inner(student):
        filtered = filter(lambda x: x[0] == x[1][0], zip(student.ans, quize))
        reduced = reduce(lambda x, y: x + y[1][1], filtered, 0)
        print(student.id, '\t', reduced)
    return inner
map(cal(QUIZE), students)

如此借助閉包(Closure)的方法,就可以維持純凈的 FP 模式啦!

走火(fn.py)

也許看了上面的 FP 寫法,你還是覺得挺啰嗦的,并沒有達到你想象中的結果,這時候就需要呈上一款語法糖利器: fn.py ! fn.py 封裝了一些常用的 FP 函數及語法糖,可以大大簡化你的代碼!

pip install fn

首先從剛剛的閉包開始,我們可以用更加 FP 的方法來解決這一問題,稱為 柯里化 ,簡單來說就是 允許接受多個參數的函數可以分次執行,每次只接受一個參數 :

from fn.func import curried  
@curried
def sum5(a, b, c, d, e):  
    return a + b + c + d + e

sum3 = sum5(1,2)  
sum4 = sum3(3,4)  
print(sum4(5))  
# 15

應用到上面的 cal 方法中:

from fn.func import curried

@curried
def cal(quize, student):  
    filtered = filter(lambda x: x[0] == x[1][0], zip(student.ans, quize))
    reduced = reduce(lambda x, y: x + y[1][1], filtered, 0)
    print(student.id, '\t', reduced)

map(cal(QUIZE), students)

在 FP 中數據通常被看作是一段數據流在一串函數的管道中傳遞,因此上面的 reduce 和 filter 其實可以合并:

reduce(lambda x, y: x + y[1][1], filter(lambda x: x[0] == x[1][0], zip(student.ans, quize)), 0)

雖然更簡略了,但是這樣會大大降低代碼的可讀性(這也是 FP 容易遭受批評的一點),為此 fn 提供了更高級的函數操作工具:

from fn import F

cal = F() >> (filter, lambda x: x[0]==x[1][0]) >> (lambda r: reduce(_+_[1][1], r, 0))  
# 計算一名學生的成績
print(cal(zip(student.ans, QUIZE)))

# 然后組合一下
@curried
def output(quize, student):  
    cal = F() >> (filter, lambda x: x[0]==x[1][0]) >> (lambda r: reduce(_+_[1][1], r, 0))
    print(student.id, '\t', cal(zip(student.ans, quize)))
map(output(QUIZE), students)

入魔(Hy)

如果你覺得上面的代碼已經足夠魔性到看起來不像是 Python 語言了,然而一旦接受了這樣的語法設定感覺也還挺不錯的。如果你興沖沖地拿去給 Lisp 或 Haskell 程序員看,則一定會被無情地鄙視:joy:,于是你痛定思痛下定決心繼續挖掘 Python 函數式編程的奧妙,那么恭喜你,組織歡迎你的加入: Hail Hydra !

哦不對,說漏了,是 Hi Hy !

Hy 是基于 Python 的 Lisp 方言,可以與 Python 代碼進行完美互嵌(如果你更偏好 PyPy,同樣也有類似的 Pixie ),除此之外你也可以把它當做一門獨立的語言來看待,它有自己的解釋器,可以當做獨立的腳本語言來使用:

pip install git+https://github.com/hylang/hy.git

首先來看一下它的基本用法,和 Python 一樣,安裝完之后可以通過 hy 命令進入 REPL 環境:

=> (print "Hy!")
Hy!  
=> (defn salutationsnm [name] (print (+ "Hy " name "!")))
=> (salutationsnm "YourName")
Hy YourName!

或者當做命令行腳本運行:

#! /usr/bin/env hy
(print "I was going to code in Python syntax, but then I got Hy.")

保存為 awesome.hy :

chmod +x awesome.hy  
./awesome.hy

接下來繼續以上面的問題為例,首先可以直接從 Python 代碼中導入:

(import data)

;; 用于 Debug 的自定義宏
;; 將可迭代對象轉化成列表后打印
(defmacro printlst [it]
    `(print (list ~it)))

(setv students data.students)
(setv quize data.QUIZE)

(defn cal [quize]
  (fn [student]
    (print student.id
      (reduce
        (fn [x y] (+ x (last (last y))))
        (filter
          (fn [x] (= (first x) (first (last x))))
          (zip student.ans quize))
        0
      )
    )
  )
)

(printl (map (cal quize) students))

如果覺得不放心,還可以直接調用最開始定義的方法將結果進行比較:

;; 假設最上面的 normal 方法保存在 fun.py 文件中
(import fun)
(.normal fun students quize)

總結

以一個簡單的數據處理問題為例,我們經歷了 Python 函數式編程從開始嘗試到“走火入魔”的整個過程。也許你還是覺得不夠過癮,想要嘗試更 純粹 的 FP 體驗,那么 Haskell 將是你最好的選擇。FP 將數據看做數據流在不同函數間傳遞,省去不必要的中間變量,保證函數的純粹性…等等這些思想在數據處理過程中還是非常有幫助的(Python 在這一領域的競爭對手 R 語言本身在語法設計上就更多地受到 Lisp 語言的影響,雖然看起來語法也比較奇怪,但這也是它比較適合用于數據處理及統計分析的原因之一)。

參考

  1. Tips ? 0x02-函數式編程
  2. Python HOWTOs ? Functional Programming HOWTO
  3. Hy's Doc
  4. fn.py

 

來自:http://blog.rainy.im/2016/12/04/fp-with-py/

 

 本文由用戶 shl198001 自行上傳分享,僅供網友學習交流。所有權歸原作者,若您的權利被侵害,請聯系管理員。
 轉載本站原創文章,請注明出處,并保留原始鏈接、圖片水印。
 本站是一個以用戶分享為主的開源技術平臺,歡迎各類分享!