繁体   English   中英

在什么情况下应该在python中使用内置的'operator'模块?

[英]In what situation should the built-in 'operator' module be used in python?

我说的是这个模块: http : //docs.python.org/library/operator.html

从文章:

operator 模块导出一组用 C 实现的函数,对应于 Python 的内部运算符。 例如,operator.add(x, y) 等价于表达式 x+y。 函数名称是用于特殊类方法的名称; 为方便起见,还提供了没有前导和尾随 __ 的变体。

我不确定我是否理解这个模块的好处或目的。

可能最流行的用法是 operator.itemgetter。 给定一个元组列表lst ,您可以按第 i 个元素排序: lst.sort(key=operator.itemgetter(i))

当然,你可以通过定义你自己的关键函数来在没有操作符的情况下做同样的事情,但是操作符模块使它稍微整洁一些。

至于其余部分,python 允许使用函数式编程风格,因此它可以出现——例如,Greg 的 reduce 示例。

你可能会争辩说:“为什么我需要operator.add当我可以做: add = lambda x, y: x+y ?” 答案是:

  1. operator.add是(我认为)稍微快一点。
  2. 它使您或以后查看它的其他人更容易理解代码。 他们不需要查找 add 的定义,因为他们知道 operator 模块的作用。
  3. operator.add是可腌制的,而lambda不是。 这意味着该函数可以保存到磁盘或在进程之间传递。

一个例子是使用reduce()函数:

>>> import operator
>>> a = [2, 3, 4, 5]
>>> reduce(lambda x, y: x + y, a)
14
>>> reduce(operator.add, a)
14

例如获取成员为元组的列表中的列,按列排序:

def item_ope():
    s = ['h', 'e', 'l', 'l', 'o']
    print operator.getitem(s, 1)
    # e
    print operator.itemgetter(1, 4)(s)
    # ('e', 'o')

    inventory = [('apple', 3), ('banana', 2), ('pear', 5), ('orange', 1)]
    get_count = operator.itemgetter(1)
    print map(get_count, inventory)
    # [3, 2, 5, 1]

    print sorted(inventory, key=get_count)
    # [('orange', 1), ('banana', 2), ('apple', 3), ('pear', 5)]

看一个更实际的例子,我们想按键或值对字典进行排序:

def dict_sort_by_value():
    dic_num = {'first': 11, 'second': 2, 'third': 33, 'Fourth': 4}

    # print all the keys
    print dic_num.keys()
    # ['second', 'Fourth', 'third', 'first']

    # sorted by value
    sorted_val = sorted(dic_num.items(), key=operator.itemgetter(1))
    # [('second', 2), ('Fourth', 4), ('first', 11), ('third', 33)]
    print sorted_val

    # sorted by key
    sorted_key = sorted(dic_num.items(), key=operator.itemgetter(0))
    print sorted_key
    # [('Fourth', 4), ('first', 11), ('second', 2), ('third', 33)]

另一个示例,当我们想要获取最大值及其在列表中的索引时:

def get_max_val_idx():
    lst = [1, 7, 3, 5, 6]
    max_val = max(lst)
    print max_val
    # 7
    max_idx = lst.index(max_val)
    print max_idx
    # 1

    # simplify it by use operator
    index, value = max(enumerate(lst), key=operator.itemgetter(1))
    print index, value
    # 1 7

更多演示如下:

import operator

def cmp_fun():
    a, b = 5, 3
    print operator.le(a, b)
    # False
    print operator.gt(a, b)
    # True


def lst_ope():
    lst = [1, 2, 3]
    print operator.indexOf(lst, 2)
    # 1
    lst1 = [1, 2, 3, 2]
    print operator.countOf(lst1, 2)
    # 2


def cal_ope():
    lst1 = [0, 1, 2, 3]
    lst2 = [10, 20, 30, 40]
    print map(operator.mul, lst1, lst2)
    # [0, 20, 60, 120]

    print sum(map(operator.mul, lst1, lst2))
    # 200

    a, b = 1, 3
    print operator.iadd(a, b)
    # 4

python 文档中查看更多信息

当您需要将函数作为参数传递给某物时,该模块很有用。 然后有两个选项:使用operator模块,或定义一个新函数(使用deflambda )。 如果你动态定义一个函数,如果你需要pickle这个函数,要么将它保存到磁盘,要么在进程之间传递它,这可能会产生问题。 虽然itemgetter是可腌制的,但动态定义的函数(使用deflambda )不是。 在以下示例中,用lambda表达式替换itemgetter将导致PicklingError

from operator import itemgetter

def sort_by_key(sequence, key):
    return sorted(sequence, key=key)

if __name__ == "__main__":
    from multiprocessing import Pool

    items = [([(1,2),(4,1)], itemgetter(1)),
             ([(5,3),(2,7)], itemgetter(0))]

    with Pool(5) as p:
        result = p.starmap(sort_by_key, items)
    print(result)

一般来说,这个模块的目的(正如上面的一些答案所暗示的那样)是为您提供简单操作的罐头函数,否则您必须自己编写并传递给高阶函数,例如sort()reduce()

例如,如果没有运算符,要对列表中的数字求和,您必须执行以下操作:

from functools import reduce

l = list(range(100))
f = lambda x, y: x + y
result = reduce(f, l)
print(result)

使用 operator 模块,您可以像这样使用它的add()函数:

from operator import add

result = reduce(add, l)
print(result)

从而避免了创建 lambda 表达式的需要。

我不记得确切的用例,但我在某处需要动态地进行一些计算,这可能会使用不同的运算符,具体取决于它的来源。

一个非常简单的例子是这样的:

import operator

def add_or_subtract(x, y, op):
    return op(x, y)

x = 3
y = 10
o = operator.add #operator.sub

add_or_subtract(x, y, o)

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM