Python的Lambda函數與排序
前幾天看到了一行求1000的階乘的Python代碼
Python代碼
- print reduce(lambda x,y:x*y, range(1, 1001))
一下子被python代碼的精簡與緊湊所折服,故對代碼進行了簡單的分析。
reduce與range都是Python的內置函數。
range(1,1001)表示生成1到1000的連續整數列表(List)。
reduce(functionA,iterableB),functionA為需要兩個變量的函數,并返回一個值。iterableB為可迭代變量,如List等。reduce函數將B中的元素從左到右依次傳入函數A中,再用函數A返回的結果替代傳入的參數,反復執行,則可將B reduce成一個單值。在此,是將1到1000的連續整數列表傳入lambda函數并用兩個數的積替換列表中的數,實際的計算過程為:(...((1×2)×3)×4)×...×1000),***的結果即1000的階乘。
下面來介紹一下lambda函數。
lambda函數是一種快速定義單行的最小函數,是從 Lisp 借用來的,可以用在任何需要函數的地方。下面的例子比較了傳統的函數與lambda函數的定義方式:
- >>> def f(x,y):
- ... return x*y
- ...
- >>> f(2,3)
- >>> g = lambda x,y: x*y
- >>> g(2,3)
可以看到,兩個函數得到的結果一樣,而對于實現簡單功能的函數來說,使用lambda函數來定義更加精簡靈活,還可以直接把函數賦值給一個變量,用變量名來表示函數名。
其實lambda函數在很多時候都是不需要賦值給一個變量的(如前文中求階乘的過程)。
使用lambda函數還有一些注意事項:
lambda 函數可以接收任意多個參數 (包括可選參數) 并且返回單個表達式的值。
lambda 函數不能包含命令,包含的表達式不能超過一個。
下面簡單演示一下如何使用lambda函數實現自定義排序。
- class People:
- age=0
- gender='male'
- def __init__(self, age, gender):
- self.age = age
- self.gender = gender
- def toString(self):
- return 'Age:'+str(self.age)+'\tGender:'+self.gender
- List=[People(21,'male'),People(20,'famale'),People(34,'male'),People(19,'famale')]
- print 'Befor sort:'
- for p in List:
- print p.toString()
- List.sort(lambda p1,p2:cmp(p1.age,p2.age))
- print '\nAfter ascending sort:'
- for p in List:
- print p.toString()
- List.sort(lambda p1,p2:-cmp(p1.age,p2.age))
- print '\nAfter descending sort:'
- for p in List:
- print p.toString()
上面的代碼定義了一個People類,并通過lambda函數,實現了對包含People類對象的列表按照People的年齡,進行升序和降序排列。運行結果如下:
- Befor sort:
- Age:21 Gender:male
- Age:20 Gender:famale
- Age:34 Gender:male
- Age:19 Gender:famale
- After ascending sort:
- Age:19 Gender:famale
- Age:20 Gender:famale
- Age:21 Gender:male
- Age:34 Gender:male
- After descending sort:
- Age:34 Gender:male
- Age:21 Gender:male
- Age:20 Gender:famale
- Age:19 Gender:famale
參考資料:
深入 Python :Dive Into Python 中文版