python 中的 operator 模块可以让它支持函数式编程。
1 计算函数
假设我们需要一个计算阶乘的函数，一般做法是使用递归。如果使用函数式编程，可以有两种方式，一种 lambda，另一种使用 operator 模块中的算术函数。我们做个比较。
首先使用 lambda 方式来实现：
|
1
2
3
|
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 函数来实现阶乘：
|
1
2
3
|
from operator import mul
def factwithmul(n):
return reduce(mul, range(1, n + 1))
|
相对来说，使用 operator 模块中的计算函数，代码更加简洁。
从源码上分析，mul 函数就是计算两个入参的乘积：
|
1
2
3
|
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 格式的城市数据，需要对其简称进行排序：
|
1
2
3
4
5
6
7
8
9
|
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)
|
运行结果：
|
1
2
3
4
5
|
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 的入参不止一个，那么它就是一个过滤函数，只把需要的列过滤出来。
比如我们只需要城市数组中的城市简称与所在坐标，就可以这么做：
|
1
2
3
|
cc_name = itemgetter(1, 3)
for city in metro_areas:
logging.info('city -> %s', cc_name(city))
|
运行结果：
|
1
2
3
4
5
|
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模块的资料请关注服务器之家其它相关文章！
原文链接：https://juejin.cn/post/6920869405177937933