python代码规范---选择分支与循环

叁歲伎倆 2022-09-06 00:18 238阅读 0赞

### 目录 ###

*   *  一： 选择分支结构
     *   *  1： 避免多层分支嵌套，能提前结束就提前结束：
         *  2: 封装那些过于复杂的逻辑判断
         *  3 ： 使用“德摩根定律”
         *  4： 自定义对象的“布尔真假”
         *  5： 在条件判断中使用 all() / any()
     *  二： 地道的循环结构
     *   *  1： enumerate() 的使用：
         *  2： 使用 product 扁平化多层嵌套循环
         *  3： 使用 islice 实现循环内隔行处理
         *  4: 使用 takewhile 替代 break 语句
         *  5： 使用生成器编写自己的修饰函数
         *  6： 按职责拆解循环体内复杂代码块

## 一： 选择分支结构 ##

### 1： 避免多层分支嵌套，能提前结束就提前结束： ###

*  优化前：
    
        def buy_fruit(nerd, store):
            """去水果店买苹果
        
            - 先得看看店是不是在营业
            - 如果有苹果的话，就买 1 个
            - 如果钱不够，就回家取钱再来
            """
            if store.is_open():
                if store.has_stocks("apple"):
                    if nerd.can_afford(store.price("apple", amount=1)):
                        nerd.buy(store, "apple", amount=1)
                        return
                    else:
                        nerd.go_home_and_get_money()
                        return buy_fruit(nerd, store)
                else:
                    raise MadAtNoFruit("no apple in store!")
            else:
                raise MadAtNoFruit("store is closed!")
 *  优化后：
    
        def buy_fruit(nerd, store):
            if not store.is_open():
                raise MadAtNoFruit("store is closed!")
        
            if not store.has_stocks("apple"):
                raise MadAtNoFruit("no apple in store!")
        
            if nerd.can_afford(store.price("apple", amount=1)):
                nerd.buy(store, "apple", amount=1)
                return
            else:
                nerd.go_home_and_get_money()
                return buy_fruit(nerd, store)

### 2: 封装那些过于复杂的逻辑判断 ###

*  **如果判断的语句太长，就用函数把条件拆开**。
    
        # 如果活动还在开放，并且活动剩余名额大于 10，为所有性别为女性，或者级别大于 3
        # 的活跃用户发放 10000 个金币
        if activity.is_active and activity.remaining > 10 and \
                user.is_active and (user.sex == 'female' or user.level > 3):
            user.add_coins(10000)
            return
    
        if activity.allow_new_user() and user.match_activity_condition():
            user.add_coins(10000)
            return

### 3 ： 使用“德摩根定律” ###

*  德摩根定律就是 `not A or not B` 等价于 `not (A and B)`。通过这样的转换

### 4： 自定义对象的“布尔真假” ###

*  通过定义魔法方法 `__len__` 和 `__bool__` ，我们可以让类自己控制想要表现出的布尔真假值，让代码变得更 pythonic。

### 5： 在条件判断中使用 all() / any() ###

*  `all(seq)`：仅当 `seq` 中所有对象都为布尔真时返回 `True`，否则返回 `False`
 *  `any(seq)`：只要 `seq` 中任何一个对象为布尔真就返回 `True`，否则返回 `False`

## 二： 地道的循环结构 ##

### 1： enumerate() 的使用： ###

*  [`enumerate()`][enumerate] 是 Python 的一个内置函数，它接收一个“可迭代”对象作为参数，然后返回一个不断生成 `(当前下标, 当前元素)` 的新可迭代对象

### 2： 使用 product 扁平化多层嵌套循环 ###

*  `product()` 可以接收多个可迭代对象，然后根据它们的笛卡尔积不断生成结果。
 *  优化前的代码：
    
        def find_twelve(num_list1, num_list2, num_list3):
            """从 3 个数字列表中，寻找是否存在和为 12 的 3 个数
            """
            for num1 in num_list1:
                for num2 in num_list2:
                    for num3 in num_list3:
                        if num1 + num2 + num3 == 12:
                            return num1, num2, num3
 *  优化后的代码：
    
        from itertools import product
        def find_twelve_v2(num_list1, num_list2, num_list3):
            for num1, num2, num3 in product(num_list1, num_list2, num_list3):
                if num1 + num2 + num3 == 12:
                    return num1, num2, num3

### 3： 使用 islice 实现循环内隔行处理 ###

*  `islice(seq, start, end, step)` 函数和数组切片操作\*（ list\[start:stop:step\] ）*有着几乎一模一样的参数。如果需要在循环内部进行隔行处理的话，只要设置第三个递进步长参数 step 值为 2 即可*（默认为 1）\*。
 *  优化前代码：
    
        def parse_titles(filename):
            """从隔行数据文件中读取 reddit 主题名称
            """
            with open(filename, 'r') as fp:
                for i, line in enumerate(fp):
                    # 跳过无意义的 '---' 分隔符
                    if i % 2 == 0:
                        yield line.strip()
 *  优化之后：
    
        from itertools import islice
        
        def parse_titles_v2(filename):
            with open(filename, 'r') as fp:
                # 设置 step=2，跳过无意义的 '---' 分隔符
                for line in islice(fp, 0, None, 2):
                    yield line.strip()

### 4: 使用 takewhile 替代 break 语句 ###

*  `takewhile(predicate, iterable)` 会在迭代 `iterable` 的过程中不断使用当前对象作为参数调用 `predicate` 函数并测试返回结果，如果函数返回值为真，则生成当前对象，循环继续。否则立即中断当前循环。
 *  优化前：
    
        for user in users:
            # 当第一个不合格的用户出现后，不再进行后面的处理
            if not is_qualified(user):
                break
 *  优化后：
    
        from itertools import takewhile
        
        for user in takewhile(is_qualified, users):

### 5： 使用生成器编写自己的修饰函数 ###

def even_only(numbers):
        for num in numbers:
            if num % 2 == 0:
                yield num
    
    
    def sum_even_only_v2(numbers):
        """对 numbers 里面所有的偶数求和"""
        result = 0
        for num in even_only(numbers):
            result += num
        return result

### 6： 按职责拆解循环体内复杂代码块 ###

[enumerate]: https://docs.python.org/3/library/functions.html#enumerate