Python函数式编程（三）操作符函数（operator）

Python开发实用教程

operator 模块提供了一套与Python的内置运算符对应的高效率函数。例如，operator.add(x, y) 与表达式 x+y 相同。 许多函数名与特殊方法名相同，只是没有双下划线。为了向后兼容性，也保留了许多包含双下划线的函数。为了表述清楚，建议使用没有双下划线的函数。
函数包含的种类有：对象的比较运算、逻辑运算、数学运算以及序列运算。
对象比较函数适用于所有的对象，函数名根据它们对应的比较运算符命名。

操作符函数

import operator
import functools
 
print(operator.le(10, 20)) #True
print(operator.add(10, 20)) #30
 
numlist = [1,2,3,4,5,6,7,8]
result = map(operator.add, numlist, numlist)
print(list(result)) #[2, 4, 6, 8, 10, 12, 14, 16]
result = functools.reduce(operator.mul, numlist)
print(result) #40320

对象属性提取attrgetter

attrgetter 类可以获取对象的属性用于 map(), stored()等高阶函数 操作

import operator
import functools
 
class Student:
    def __init__(self, name, age):
        self.name = name
        self.age = age
 
    def __repr__(self):
        return f'({self.name=}, {self.age=})'
 
slist = [Student('Alice', 15), Student('Bibe', 16), Student('Chen', 12), Student('Jack', 13)]
 
#按年龄排序
print(sorted(slist, key=operator.attrgetter('age')))
#按姓名排序
print(sorted(slist, key=operator.attrgetter('name')))
 
‘’'
[(self.name='Chen', self.age=12), (self.name='Jack', self.age=13), (self.name='Alice', self.age=15), (self.name='Bibe', self.age=16)]
[(self.name='Alice', self.age=15), (self.name='Bibe', self.age=16), (self.name='Chen', self.age=12), (self.name='Jack', self.age=13)]
‘''

对象方法提取methodcaller

operator.methodcaller(name, /, *args, **kwargs)

返回一个在操作数上调用 name 方法的可调用对象。 如果给出额外的参数和/或关键字参数，它们也将被传给该方法。 例如：

• 在 f = methodcaller('name') 之后，调用 f(b) 将返回 b.name()。
• 在 f = methodcaller('name', 'foo', bar=1) 之后，调用 f(b) 将返回 b.name('foo', bar=1)。

import operator
import functools
 
class Student:
    def __init__(self, name, age):
        self.name = name
        self.age = age
 
    def __repr__(self):
        return f'({self.name=}, {self.age=})'
 
    def isPupil(self):
        if self.age >=6 and self.age < 12:
            return True
 
        return False
 
slist = [Student('Alice', 6), Student('Bibe', 8), Student('Chen', 12), Student('Jack', 9)]
 
print(list(filter(operator.methodcaller('isPupil'), slist)))
#[(self.name='Alice', self.age=6), (self.name='Bibe', self.age=8), (self.name='Jack', self.age=9)]
 

下标获取itemgetter

operator.itemgetter(item)

operator.itemgetter(*items)

返回一个使用操作数的 __getitem__() 方法从操作数中获取 item 的可调用对象。 如果指定了多个条目，则返回一个查找值的元组。 例如：
在 f = itemgetter(2) 之后，调用 f(r) 将返回 r[2]。
在 g = itemgetter(2, 5, 3) 之后，调用 g(r) 将返回 (r[2], r[5], r[3])。

>>>itemgetter(1)('ABCDEFG')
'B'
>>>itemgetter(1, 3, 5)('ABCDEFG')
('B', 'D', 'F')
>>>itemgetter(slice(2, None))('ABCDEFG')
'CDEFG'
>>>soldier = dict(rank='captain', name='dotterbart')
>>>itemgetter('rank')(soldier)
'captain'

使用 itemgetter() 从元组的记录中提取特定字段的例子：

>>>inventory = [('apple', 3), ('banana', 2), ('pear', 5), ('orange', 1)]
>>>getcount = itemgetter(1)
>>>list(map(getcount, inventory))
[3, 2, 5, 1]
>>>sorted(inventory, key=getcount)
[('orange', 1), ('banana', 2), ('apple', 3), ('pear', 5)]

将运算符映射到函数