欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Python的Lambda函数用法详解

程序员文章站 2022-06-03 11:07:25
在python中有两种函数,一种是def定义的函数,另一种是lambda函数,也就是大家常说的匿名函数。今天我就和大家聊聊lambda函数,在python编程中,大家习惯将...

在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函数的用法上面也讲到了匿名函数的优点,那它到底有哪些用处呢?

(1)直接赋给一个变量,然后再像一般函数那样调用

c=lambda x,y,z:x*y*z 
c(2,3,4) 
 
24 

当然,也可以在函数后面直接传递实参

(lambda x:x**2)(3) 
9 

(2)将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 

(3)嵌套使用将lambda函数嵌套到普通函数中,lambda函数本身做为return的值

def increment(n): 
  return lambda x:x+n 
 
f=increment(4) 
f(2) 
6 

(4)字符串联合,有默认值,也可以用x=(lambda...)这种格式

x=(lambda x='boo',y='too',z='z00':x+y+z) 
print(x('foo')) 
 
'footoozoo' 

(5)在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() 

这段代码还是挺有意思的,希望小伙伴们可以复制粘贴运行一下哈。(6)判断字符串是否以某个字母开头有

names = ['anne', 'amy', 'bob', 'david', 'carrie', 'barbara', 'zach'] 
b_name= filter(lambda x: x.startswith('b'),names) 
print(b_name) 
['bob', 'barbara'] 

(7)求两个列表元素的和

a = [1,2,3,4] 
b = [5,6,7,8] 
print(list(map(lambda x,y:x+y, a,b))) 
[6,8,10,12] 

(8)求字符串每个单词的长度

sentence = "welcome to beijing!" 
words = sentence.split() 
lengths = map(lambda x:len(x),words) 
print(list(lengths)) 
[7,2,8] 

总结

对于lambda函数的使用在python社区一直存在争议,支持一方认为lambad函数的使用,使得python代码更加紧凑,更加pythonic;反对方则认为python的lambda函数限制多多,最严重的当属于它只能由一条表达式组成,用多了以后反而使得程序看起来不那么清晰。

但我个人觉得作为一个python的初级用户,lambda函数确实让编码更简洁,更省事,一句话好用就行,其余的看神仙打架吧!

以上所述是小编给大家介绍的python的lambda函数用法详解,希望对大家有所帮助