美文网首页
Python lambda 用法

Python lambda 用法

作者: Paycation | 来源:发表于2018-05-17 22:22 被阅读269次

    问题:
    请使用 sorted 和 lambda 将下列字典转成名字为元素的列表,并且按照年龄升序排序。

    ages = {'michael': 25, 'brandy': 26, 'sue': 29, 'aria':22}
    

    需要用 sorted 的 key 参数。用于传入一个函数,这个函数的作用是将序列的每一项输入这个函数,并根据返回值来排序。

    # get 方法用于返回 dict 对象的一个键对应的值。
    m = ages.get('michael') # m = 25
    out = sorted(ages, key=ages.get) # 用 25,26,29,22排序
    # sorted 会自动把各种 iterator 
    转成 list 然后再排序。
    # 输出:
    ['aria', 'brandy', 'michael', 'sue']
    

    好,现在我们用 lambda 代替 get。lambda 其实就是一个函数构造器,语法是:
    lamda x:do(x) 冒号前面是参数,冒号后面是输出。我们要做的很简单,就是输入键,再从字典里取值。注意 key 的函数的输入是 iterator 的每一项。i 就是字典的键,输出 ages[i] 字典的值。

    out = sorted(ages, key=lambda i:ages[i]) 
    

    更多例子

    map

    使用 map 函数,可以把自变量映射到函数上,生成一个 iterator。

    for i in map(lambda i:i+1, [1, 2, 3]):
        print(i)
    # 输出
    2
    3
    4
    

    lambda i:i+1 的功能就是:给某数加1,连名字都懒得取。特别是遇到只用一次的函数,可以就用 lambda 构造。

    max / min

    max 和 min 也有类似的用法。

    max(ages, key=lambda i:ages[i])
    # 输出
    sue
    

    filter

    从 iterator 中筛选返回值为 True 的元素。
    比如我只要 1 到 100 内 3 的倍数:

    filter(lambda n: n%3==0, range(1, 101))
    

    PS:也可以用 list comprehension 写。

    [x for x in range(1, 101) if x % 3 == 0]
    

    两者不同之处是,上面是 iterator,而下面则是具体的 list 对象。

    小结

    总得来说,lambda 主要还是用在 key 参数里面。其他的情况我还没怎么遇到。

    相关文章

      网友评论

          本文标题:Python lambda 用法

          本文链接:https://www.haomeiwen.com/subject/whwidftx.html