Python的lambda函数与排序

2010-03-02 15:02 2809人阅读评论(0) 收藏举报

前几天看到了一行求1000的阶乘的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函数的定义方式：

1 >>> def f ( x ,y):
2 ... return x * y
3 ...
4 >>> f ( 2,3 )
5 6
6 >>> g = lambda x ,y: x * y
7 >>> g ( 2,3 )
8 6

可以看到，两个函数得到的结果一样，而对于实现简单功能的函数来说，使用lambda函数来定义更加精简灵活，还可以直接把函数赋值给一个变量，用变量名来表示函数名。

其实lambda函数在很多时候都是不需要赋值给一个变量的（如前文中求阶乘的过程）。

使用lambda函数还有一些注意事项：

lambda 函数可以接收任意多个参数 (包括可选参数) 并且返回单个表达式的值。

lambda 函数不能包含命令，包含的表达式不能超过一个。

下面简单演示一下如何使用lambda函数实现自定义排序。

01 class People :
02     age = 0
03     gender = 'male'
04
05     def __init__ ( self , age , gender ):
06         self . age = age
07         self . gender = gender
08     def toString ( self ):
09         return 'Age:' + str ( self . age ) + ' /t Gender:' + self . gender
10
11 List = [ People ( 21 , 'male' ), People ( 20 , 'famale' ), People ( 34 , 'male' ), People ( 19 , 'famale' )]
12 print 'Befor sort:'
13 for p in List :
14     print p . toString ()
15
16 List . sort ( lambda p1 , p2 : cmp ( p1 . age , p2 . age ))
17 print ' /n After ascending sort:'
18 for p in List :
19     print p . toString ()
20
21 List . sort ( lambda p1 , p2 : - cmp ( p1 . age , p2 . age ))
22 print ' /n After descending sort:'
23 for p in List :
24     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

lambda 语句被用来创建新的函数对象，并且在运行时返回它们。

例15.2 使用lambda形式

#!/usr/bin/python # Filename: lambda.py

def make_repeater (n):
return lambda s: s*n

twice = make_repeater( 2 )

print twice( 'word' )
print twice( 5 )

（源文件：code/lambda.py）

输出

$ python lambda.py wordword 10

它如何工作

这里，我们使用了make_repeater 函数在运行时创建新的函数对象，并且返回它。lambda 语句用来创建函数对象。本质上，lambda 需要一个参数，后面仅跟单个表达式作为函数体，而表达式的值被这个新建的函数返回。注意，即便是print 语句也不能用在 lambda形式中，只能使用表达式。