细说Python的lambda函数用法
在Python中有两种函数,一种是def定义的函数,另一种是lambda函数,也就是大家常说的匿名函数。今天我就和大家聊聊lambda函数,在Python编程中,大家习惯将其称为表达式。
1.为什么要用lambda函数?
先举一个例子:将一个列表里的每个元素都平方。
先用def来定义函数,代码如下
| def sq(x):
return x*x
map(sq,[y for y in range(10)])
|
再用lambda函数来编写代码
| map(lambda x: x*x,[y for y in range(10)])
|
从这个简单的例子,我们可以看出,用lambda函数首先减少了代码的冗余,其次,用lambda函数,不用费神地去命名一个函数的名字,可以快速的实现某项功能,最后,lambda函数使代码的可读性更强,程序看起来更加简洁。
从上面这个简单的例子,也可以看出来lambda函数的语法是唯一的,其形式如下:
| lambda argument_list:expersion
|
语法中的argument_list是参数列表,它的结构与Python中函数(function)的参数列表是一样的,例如
| a,b
a=1,b=2
*args
**kwargs
a,b=1,*args
空
....
|
语法中的expression是一个关于参数的表达式,表达式中出现的参数需要在argument_list中有定义,并且表达式只能是单行的。比如以下的一些合法的表达式
| 1
None
a+b
sum(a)
1 if a ;10 else 0
......
|
除了上面提到的lambda函数的优点外,我看有的文章说用lambda函数会提高效率,那究竟是不是呢?我们写一段代码来验证一下
| import time
# 测试的Def函数
def square1(n):
return n ** 2
# 测试的Lambda函数
square2 = lambda n: n ** 2
print(time.time())
# 使用Def函数
i = 0
while i ; 1000000000:
square1(100)
i += 1
print(time.time())
# 使用lambda函数
i = 0
while i ; 1000000000:
square2(100)
i += 1
print(time.time())
1413272496.27
1413272703.05 (Def 函数:207s)
1413272904.49 (Lambda函数:201s)
|
从上面可以看出,两种的所需的时间差不多,效率丝毫不受影响。
2.lambad函数的用法上面也讲到了匿名函数的优点,那它到底有哪些用处呢?
直接赋给一个变量,然后再像一般函数那样调用
| c=lambda x,y,z:x*y*z
c(2,3,4)
24
|
在函数后面直接传递实参
将lambda函数作为参数传递给其他函数比如说结合
map、filter、sorted、reduce等一些Python内置函数使用,下面举例说明。
| fliter(lambda x:x%3==0,[1,2,3,4,5,6])
[3,6]
squares = map(lambda x:x**2,range(5)
print(lsit(squares))
[0,1,4,9,16]
|
与sorted函数结合使用,比如:创建由元组构成的列表:
| a=[('b',3),('a',2),('d',4),('c',1)]
|
按照第一个元素排序
| sorted(a,key=lambda x:x[0])
[('a',2),('b',3),('c',1),('d',4)]
|
按照第二个元素排序
| sorted(a,key=lambda x:x[1])
[('c',1),('a',2),('b',3),('d',4)]
|
与reduce函数结合使用
| from functools import reduce
print(reduce(lambda a,b:'{},{}'.format(a,b),[1,2,3,4,5,6,7,8,9]))
1,2,3,4,5,6,7,8,9
|
嵌套使用将lambda函数嵌套到普通函数中,lambda函数本身做为return的值
| def increment(n):
return lambda x:x+n
f=increment(4)
f(2)
6
|
字符串联合,有默认值,也可以用x=(lambda…)这种格式
| x=(lambda x='Boo',y='Too',z='Z00':x+y+z)
print(x('Foo'))
'FooTooZoo'
|
在tkinter中定义内联的callback函数
| import sys
from tkinter import Button,mainloop
x=Button(text='Press me',command=(lambda :sys.stdout.write('Hello,World\n')))
x.pack()
x.mainloop()
|
判断字符串是否以某个字母开头有
| Names = ['Anne', 'Amy', 'Bob', 'David', 'Carrie', 'Barbara', 'Zach']
B_Name= filter(lambda x: x.startswith('B'),Names)
print(B_Name)
['Bob', 'Barbara']
|
求两个列表元素的和
| a = [1,2,3,4]
b = [5,6,7,8]
print(list(map(lambda x,y:x+y, a,b)))
[6,8,10,12]
|
求字符串每个单词的长度
| sentence = "Welcome To Beijing!"
words = sentence.split()
lengths = map(lambda x:len(x),words)
print(list(lengths))
[7,2,8]
|