Python代码精简技巧：列表推导式、条件表达式与Lambda函数实战

Python代码精简技巧：列表推导式、条件表达式与Lambda函数实战

在使用Jupyter Notebook进行开发时，往往感觉代码行数珍贵，因此有必要总结一些常用的高效简写方法。

1. for循环的简写形式

首先通过一个具体示例来展示其效果：

y = [1,2,3,4,5,6]
[(i*2) for i in y ]

最终输出结果为 [2, 4, 6, 8, 10, 12]

1.1 单层for循环的简写

单层for循环的简写语法如下（注意必须使用方括号）：

• [ 对元素i的操作 for i in 列表 ]

该写法等价于以下传统代码结构：

• for i in 列表:
• 对i执行操作

1.2 多层for循环的简写

当涉及两层for循环时，简写格式为：

• [对元素i的操作 for 单个元素 in 列表 for i in 单个元素]

以下是一个简单的应用示例：

y_list = ['assss','dvv']
[print(i) for y in y_list for i in y]

得到的输出结果为：a s s s s d v v

其对应的传统循环写法如下：

y_list = ['assss','dvv']
for y in y_list:
    for i in y:
        print(i) 

2. if条件的简写形式

其基本语法格式为：

• 条件为True时的逻辑 if 判断条件 else 条件为False时的逻辑

具体示例如下：

y = 0
x = y+3 if y > 3 else y-1

此时设定 x = -1

由于 y = 0，判断 y>3 不成立，因此执行了False分支的逻辑（y-1），最终x的值为 -1

2.1 for循环与if条件的结合简写

举例说明：

x = [1,2,3,4,5,6,7]
[print(i) for i in x if i > 3 ]

程序将输出：4 5 6 7

需要注意的是，使用这种简写方式无法对判断为False的对象执行额外操作。

因此，其通用模板为：

• [判断为True的元素i的操作 for i in 列表 if i的判断条件]

3. 匿名函数lambda的使用

匿名函数的定义语法如下：

• lambda 参数: 表达式

示例如下：

x = 3
(lambda k: k+3)(x)

输出结果为 6

这是一个较为简单的匿名函数表达式。在实际开发中，匿名函数常与其他函数结合使用，主要作为参数传递。例如：

比如：map()函数配合匿名函数

map(函数名, 列表) 的作用是对列表中的每个元素执行该函数的操作，具体示例如下：

list(map(lambda x: x * 2, [1, 2, 3, 4, 5]))

输出结果为： [2, 4, 6, 8, 10]

Python代码if-else简写优化

Python代码是逐行执行的，减少代码行数通常能提高执行效率。优化语句写法不仅能提升代码可读性，还能使代码更加简洁。例如，一个常用的判断赋值语句传统写法通常需要4行代码。

if b>0:
    a=b
else:
    a=0

实际上，使用一行代码即可完成上述操作：

#如果b的值大于0，将b直接赋值给a，否则将a设置为0
a=b if b>0 else 0

以上为个人经验总结，希望能为大家提供参考，也希望大家多多支持本站。