python避免代码过度嵌套

代码嵌套让代码变得难以阅读,尤其当嵌套层次超过3层以后,嵌套的太多不符合python扁平胜于嵌套的编程理念,那么怎么才能减少代码的嵌套呢?

1. 使用生成式

tmp = [4, 2, 1, 5, 7]

从列表tmp中找出大于3的数据,可以使用for循环来完成

tmp = [4, 2, 1, 5, 7]

lst = []
for item in tmp:
    if item > 3:
        lst.append(item)

这样编写代码没有问题,但不够pythonic,你可以使用列表生成式让代码更简单,而且没有嵌套

tmp = [4, 2, 1, 5, 7]

lst = [item for item in tmp if item > 3]
print(lst)

2. 三元表达式

python没有三元表达式,但可以借助if else 来实现

a = 5

if a > 3:
    b = 4
else:
    b = 8

这段代码里,a的值决定了b的值, 我使用if else 语句来编写,出现了嵌套,为了减少嵌套,可以用下面的方法

a = 5

b = 4 if a > 3 else 8

代码瞬间变得清爽了许多

3. 优化条件表达式

lst1 = [1, 2, 6, 4, 7, 8, 9]
lst2 = [14, 13, 11, 8, 6]

编写算法,从两个列表中各取出一个数值,如果两个数值都大于5且他们的和是15,则输出这两个数值

下面向你展示一个非常糟糕的写法

lst1 = [1, 2, 6, 4, 7, 8, 9]
lst2 = [14, 13, 11, 8, 6]

for item1 in lst1:
    for item2 in lst2:
        if item1 > 5:
            if item2 > 5:
                if item1 + item2 == 15:
                    print(item1, item2)

虽然程序的逻辑是正确的,但是形成了太多的嵌套语句,使得代码难以阅读,下面是修改后的

lst1 = [1, 2, 6, 4, 7, 8, 9]
lst2 = [14, 13, 11, 8, 6]

for item1 in lst1:
    for item2 in lst2:
        if item1 > 5 and item2 > 5 and item1 + item2 == 15:
            print(item1, item2)

扫描关注, 与我技术互动

QQ交流群: 211426309

加入知识星球, 每天收获更多精彩内容

分享日常研究的python技术和遇到的问题及解决方案