Python 中Operator模塊的使用

Python 中的 Operator 模塊可以讓它支持函數式編程。

1 計算函數

假設我們需要一個計算階乘的函數,一般做法是使用遞歸。如果使用函數式編程,可以有兩種方式,一種 lambda,另一種使用 Operator 模塊中的算術函數。我們做個比較。

首先使用 lambda 方式來實現:

from functools import reduce
def fact(n):
  return reduce(lambda a, b: a * b, range(1, n + 1))

這裡用到瞭 reduce 方法,reduce() 函數語法:reduce(function, iterable[, initializer])。其中的 function 函數有兩個參數。reduce() 函數會先對集合中的第 1、2 個元素進行 function 函數處理,得到的結果再與第三個元素進行 function 函數處理,最後得到一個結果1。

reduce() 函數中的 function 使用 lambda 表達式。第二個入參 iterable 使用 range() 方法來生成。

接著使用 Operator 模塊中的 mul 函數來實現階乘:

from operator import mul
def factWithMul(n):
  return reduce(mul, range(1, n + 1))

相對來說,使用 Operator 模塊中的計算函數,代碼更加簡潔。

從源碼上分析,mul 函數就是計算兩個入參的乘積:

def mul(a, b):
  "Same as a * b."
  return a * b

Operator 模塊內部還定義瞭很多類似的計算函數,比如:

2 過濾函數

operator 模塊中還有一類函數能夠起到條件過濾的作用。

首先介紹 itemgetter 函數。假設定義瞭一個 f 函數,f = itemgetter(2),那麼如果調用 f(r),實際會返回 r[2]。也就是說,傳入的入參 r,最終返回的隻是 r 的索引值為 2 (從 1 開始計算)那一部分。

itemgetter 函數應用場景是:根據元組的某個字段對元組列表進行排序。比如我們有一段 JSON 格式的城市數據,需要對其簡稱進行排序:

from operator import itemgetter

metro_areas = [('Tokyo', 'JP', 36.933, (35.689722, 139.691667)), ('Delhi NCR', 'IN', 21.935, (28.613889, 77.208889)),
        ('Mexico City', 'MX', 20.142, (19.433333, -99.133333)),
        ('New York-Newark', 'US', 20.104, (40.808611, -74.020386)),
        ('Sao Paulo', 'BR', 19.649, (-23.547778, -46.635833)),
        ]
for city in sorted(metro_areas, key=itemgetter(1)):
  logging.info('city -> %s', city)

運行結果:

INFO - city -> ('Sao Paulo', 'BR', 19.649, (-23.547778, -46.635833))
INFO - city -> ('Delhi NCR', 'IN', 21.935, (28.613889, 77.208889))
INFO - city -> ('Tokyo', 'JP', 36.933, (35.689722, 139.691667))
INFO - city -> ('Mexico City', 'MX', 20.142, (19.433333, -99.133333))
INFO - city -> ('New York-Newark', 'US', 20.104, (40.808611, -74.020386))

城市數據數組中的每一個元素,拆分來看,就是元組形式。 itemgetter 函數取出元組的第二個值作為 sorted 函數的排序依據。

如果傳入 itemgetter 的入參不止一個,那麼它就是一個過濾函數,隻把需要的列過濾出來。

比如我們隻需要城市數組中的城市簡稱與所在坐標,就可以這麼做:

cc_name = itemgetter(1, 3)
for city in metro_areas:
logging.info('city -> %s', cc_name(city))

運行結果:

INFO - city -> ('JP', (35.689722, 139.691667))
INFO - city -> ('IN', (28.613889, 77.208889))
INFO - city -> ('MX', (19.433333, -99.133333))
INFO - city -> ('US', (40.808611, -74.020386))
INFO - city -> ('BR', (-23.547778, -46.635833))

以上就是Python 中Operator模塊的使用的詳細內容,更多關於Python 中Operator模塊的資料請關註WalkonNet其它相關文章!