Python lambda表达式用法

程序员文章站 2024-02-26 19:06:34

...

python lambda表达式简单用法

lambda表达式简单用法

lambda是什么？

看个例子：

g = lambda x:x+1

看一下执行的结果：

g(1)
>>>2

g(2)

>>>3

当然，你也可以这样使用：

python lambda x:x+1(1)

>>>2

可以这样认为,lambda作为一个表达式，定义了一个匿名函数，上例的代码x为入口参数，x+1为函数体，用函数来表示为：

 def g(x):
     return x+1

非常容易理解，在这里lambda简化了函数定义的书写形式。是代码更为简洁，但是使用函数的定义方式更为直观，易理解。

Python中，也有几个定义好的全局函数方便使用的，filter, map, reduce

>>> foo = [2, 18, 9, 22, 17, 24, 8, 12, 27]

>>> print filter(lambda x: x % 3 == 0, foo)
[18, 9, 24, 12, 27]

>>> print map(lambda x: x * 2 + 10, foo)
[14, 46, 28, 54, 44, 58, 26, 34, 64]

>>> print reduce(lambda x, y: x + y, foo)
139

上面例子中的map的作用，非常简单清晰。但是，Python是否非要使用lambda才能做到这样的简洁程度呢？在对象遍历处理方面，其实Python的for…in…if语法已经很强大，并且在易读上胜过了lambda。

比如上面map的例子，可以写成：

print [x * 2 + 10 for x in foo]

非常的简洁，易懂。

filter的例子可以写成：

print [x for x in foo if x % 3 == 0]

同样也是比lambda的方式更容易理解。

上面简要介绍了什么是lambda,下面介绍为什么使用lambda,看一个例子（来自apihelper.py)：

processFunc = collapse and (lambda s: " ".join(s.split())) or (lambda s: s)
　　在Visual Basic，你很有可能要创建一个函数，接受一个字符串参数和一个 collapse 参数，并使用 if 语句确定是否压缩空白，然后再返回相应的值。这种方式是低效的，因为函数可能需要处理每一种可能的情况。每次你调用它，它将不得不在给出你所想要的东西之前，判断是否要压缩空白。在 Python 中，你可以将决策逻辑拿到函数外面，而定义一个裁减过的 lambda 函数提供确切的 (唯一的) 你想要的。这种方式更为高效、更为优雅，而且很少引起那些令人讨厌 (哦，想到那些参数就头昏) 的错误。

通过此例子，我们发现，lambda的使用大量简化了代码，使代码简练清晰。但是值得注意的是，这会在一定程度上降低代码的可读性。

习条件运算时，对于简单的 if else 语句，可以使用三元运算来表示，即：

普通条件语句

if 1 == 1:
    name = 'wupeiqi'
else:
    name = 'alex'

三元运算

name = 'wupeiqi' if 1 == 1 else 'alex'

对于简单的函数，也存在一种简便的表示方式，即：lambda表达式

---------------------------------------普通函数-------------------------------------
定义函数（普通方式）

def func(arg):
    return arg + 1

执行函数

result = func(123)

---------------------------------------lambda----------------------------------------

定义函数（lambda表达式）

my_lambda = lambda arg : arg + 1

执行函数

result = my_lambda(123)

lambda存在意义就是对简单函数的简洁表示

lambda表达式在sort函数中的使用

假如a是一个由元组构成的列表，对该列表进行排序时，我们需要用到参数key，也就是关键词，如下面代码所示，lambda是一个匿名函数，是固定写法；x表示匿名函数的输入，即列表中的一个元素，在这里，表示一个元组，x只是临时起的一个名字，你可以使用任意的名字；x[0]表示匿名函数的输出，即元组里的第一个元素，即key = x[0]；所以这句命令的意思就是按照列表中第一个元素进行排序。

>>> a = [('b', 4), ('a', 12), ('d', 7), ('h', 6), ('j', 3)]
>>> a.sort(key=lambda x: x[0])
>>> print(a)
[('a', 12), ('b', 4), ('d', 7), ('h', 6), ('j', 3)]