細說(shuō)Python的lambda函數用法,建議收藏
在Python中有兩種函數,一種是def定義的函數,另一種是lambda函數,也就是大家常說(shuō)的匿名函數。今天我就和大家聊聊lambda函數,在Python編程中,大家習慣將其稱(chēng)為表達式。
1.為什么要用lambda函數?
先舉一個(gè)例子:將一個(gè)列表里的每個(gè)元素都平方。
先用def來(lái)定義函數,代碼如下
def sq(x):
return x*x
map(sq,[y for y in range(10)])
再用lambda函數來(lái)編寫(xiě)代碼
map(lambda x: x*x,[y for y in range(10)])
從這個(gè)簡(jiǎn)單的例子,我們可以看出,用lambda函數首先減少了代碼的冗余,其次,用lambda函數,不用費神地去命名一個(gè)函數的名字,可以快速的實(shí)現某項功能,最后,lambda函數使代碼的可讀性更強,程序看起來(lái)更加簡(jiǎn)潔。
從上面這個(gè)簡(jiǎn)單的例子,也可以看出來(lái)lambda函數的語(yǔ)法是唯一的,其形式如下:
lambda argument_list:expersion
語(yǔ)法中的argument_list是參數列表,它的結構與Python中函數(function)的參數列表是一樣的,例如
a,b
a=1,b=2
*args
**kwargs
a,b=1,*args
空
....
語(yǔ)法中的expression是一個(gè)關(guān)于參數的表達式,表達式中出現的參數需要在argument_list中有定義,并且表達式只能是單行的。比如以下的一些合法的表達式
1
None
a+b
sum(a)
1 if a >10 else 0
......
除了上面提到的lambda函數的優(yōu)點(diǎn)外,我看有的文章說(shuō)用lambda函數會(huì )提高效率,那究竟是不是呢?我們寫(xiě)一段代碼來(lái)驗證一下
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)
從上面可以看出,兩種的所需的時(shí)間差不多,效率絲毫不受影響。
2.lambad函數的用法上面也講到了匿名函數的優(yōu)點(diǎn),那它到底有哪些用處呢?
(1)直接賦給一個(gè)變量,然后再像一般函數那樣調用
c=lambda x,y,z:x*y*z
c(2,3,4)
24
當然,也可以在函數后面直接傳遞實(shí)參
(lambda x:x**2)(3)
9
(2)將lambda函數作為參數傳遞給其他函數比如說(shuō)結合map、filter、sorted、reduce等一些Python內置函數使用,下面舉例說(shuō)明。
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函數結合使用,比如:創(chuàng )建由元組構成的列表:
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函數結合使用
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)字符串聯(lián)合,有默認值,也可以用x=(lambda...)這種格式
x=(lambda x='Boo',y='Too',z='Z00':x+y+z)
print(x('Foo'))
'FooTooZoo'
(5)在tkinter中定義內聯(lián)的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)判斷字符串是否以某個(gè)字母開(kāi)頭有
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è)單詞的長(cháng)度
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函數限制多多,最嚴重的當屬于它只能由一條表達式組成,用多了以后反而使得程序看起來(lái)不那么清晰。