Python lambda 用法

问题:
请使用 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 用法)