細(xì)說Python的Lambda函數(shù)用法,建議收藏
在Python中有兩種函數(shù),一種是def定義的函數(shù),另一種是lambda函數(shù),也就是大家常說的匿名函數(shù)。今天我就和大家聊聊lambda函數(shù),在Python編程中,大家習(xí)慣將其稱為表達(dá)式。
1.為什么要用lambda函數(shù)?
先舉一個(gè)例子:將一個(gè)列表里的每個(gè)元素都平方。
先用def來定義函數(shù),代碼如下
- def sq(x):
- return x*x
- map(sq,[y for y in range(10)])
再用lambda函數(shù)來編寫代碼
- map(lambda x: x*x,[y for y in range(10)])
從這個(gè)簡單的例子,我們可以看出,用lambda函數(shù)首先減少了代碼的冗余,其次,用lambda函數(shù),不用費(fèi)神地去命名一個(gè)函數(shù)的名字,可以快速的實(shí)現(xiàn)某項(xiàng)功能,最后,lambda函數(shù)使代碼的可讀性更強(qiáng),程序看起來更加簡潔。
從上面這個(gè)簡單的例子,也可以看出來lambda函數(shù)的語法是唯一的,其形式如下:
- lambda argument_list:expersion
語法中的argument_list是參數(shù)列表,它的結(jié)構(gòu)與Python中函數(shù)(function)的參數(shù)列表是一樣的,例如
- a,b
- a=1,b=2
- *args
- **kwargs
- a,b=1,*args
- 空
- ....
語法中的expression是一個(gè)關(guān)于參數(shù)的表達(dá)式,表達(dá)式中出現(xiàn)的參數(shù)需要在argument_list中有定義,并且表達(dá)式只能是單行的。比如以下的一些合法的表達(dá)式
- 1
- None
- a+b
- sum(a)
- 1 if a >10 else 0
- ......
除了上面提到的lambda函數(shù)的優(yōu)點(diǎn)外,我看有的文章說用lambda函數(shù)會(huì)提高效率,那究竟是不是呢?我們寫一段代碼來驗(yàn)證一下
- import time
- # 測試的Def函數(shù)
- def square1(n):
- return n ** 2
- # 測試的Lambda函數(shù)
- square2 = lambda n: n ** 2
- print(time.time())
- # 使用Def函數(shù)
- i = 0
- while i < 1000000000:
- square1(100)
- i += 1
- print(time.time())
- # 使用lambda函數(shù)
- i = 0
- while i < 1000000000:
- square2(100)
- i += 1
- print(time.time())
- 1413272496.27
- 1413272703.05 (Def 函數(shù):207s)
- 1413272904.49 (Lambda函數(shù):201s)
從上面可以看出,兩種的所需的時(shí)間差不多,效率絲毫不受影響。
2.lambad函數(shù)的用法上面也講到了匿名函數(shù)的優(yōu)點(diǎn),那它到底有哪些用處呢?
(1)直接賦給一個(gè)變量,然后再像一般函數(shù)那樣調(diào)用
- c=lambda x,y,z:x*y*z
- c(2,3,4)
- 24
當(dāng)然,也可以在函數(shù)后面直接傳遞實(shí)參
- (lambda x:x**2)(3)
- 9
(2)將lambda函數(shù)作為參數(shù)傳遞給其他函數(shù)比如說結(jié)合map、filter、sorted、reduce等一些Python內(nèi)置函數(shù)使用,下面舉例說明。
- 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函數(shù)結(jié)合使用,比如:創(chuàng)建由元組構(gòu)成的列表:
- a=[('b',3),('a',2),('d',4),('c',1)]
按照第一個(gè)元素排序
- sorted(a,key=lambda x:x[0])
- [('a',2),('b',3),('c',1),('d',4)]
按照第二個(gè)元素排序
- sorted(a,key=lambda x:x[1])
- [('c',1),('a',2),('b',3),('d',4)]
與reduce函數(shù)結(jié)合使用
- 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函數(shù)嵌套到普通函數(shù)中,lambda函數(shù)本身做為return的值
- def increment(n):
- return lambda x:x+n
- f=increment(4)
- f(2)
- 6
(4)字符串聯(lián)合,有默認(rèn)值,也可以用x=(lambda...)這種格式
- x=(lambda x='Boo',y='Too',z='Z00':x+y+z)
- print(x('Foo'))
- 'FooTooZoo'
(5)在tkinter中定義內(nèi)聯(lián)的callback函數(shù)
- import sys
- from tkinter import Button,mainloop
- x=Button(text='Press me',command=(lambda :sys.stdout.write('Hello,World\n')))
- x.pack()
- x.mainloop()
這段代碼還是挺有意思的,希望小伙伴們可以復(fù)制粘貼運(yùn)行一下哈。(6)判斷字符串是否以某個(gè)字母開頭有
- Names = ['Anne', 'Amy', 'Bob', 'David', 'Carrie', 'Barbara', 'Zach']
- B_Name= filter(lambda x: x.startswith('B'),Names)
- print(B_Name)
- ['Bob', 'Barbara']
(7)求兩個(gè)列表元素的和
- 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)求字符串每個(gè)單詞的長度
- sentence = "Welcome To Beijing!"
- words = sentence.split()
- lengths = map(lambda x:len(x),words)
- print(list(lengths))
- [7,2,8]
總結(jié)
對(duì)于lambda函數(shù)的使用在Python社區(qū)一直存在爭議,支持一方認(rèn)為lambad函數(shù)的使用,使得Python代碼更加緊湊,更加Pythonic;反對(duì)方則認(rèn)為Python的lambda函數(shù)限制多多,最嚴(yán)重的當(dāng)屬于它只能由一條表達(dá)式組成,用多了以后反而使得程序看起來不那么清晰。
但我個(gè)人覺得作為一個(gè)Python的初級(jí)用戶,lambda函數(shù)確實(shí)讓編碼更簡潔,更省事,一句話好用就行,其余的看神仙打架吧!